diff --git a/tests/unit/test_fused_router.py b/tests/unit/test_fused_router.py
index 3250cef4..4fc3bd5f 100644
--- a/tests/unit/test_fused_router.py
+++ b/tests/unit/test_fused_router.py
@@ -40,10 +40,10 @@ def test_fused_router():
     gate_lin = Nvfp4Linear(in_features=K, out_features=N, device=device)
     gate_lin.fp4 = [w_fp4.contiguous()]
     gate_lin.sf = [w_sf.contiguous()]
-    gate_lin.gsb = [ws2_val]
+    gate_lin.gsb = None
+    gate_lin.gs = [1.0]  # default global scale
+    gate_lin.ws2 = [ws2_val]
     gate_lin._activation_global_scale = None  # set at runtime
-    gate_lin._ensure_stacked = lambda *a, **kw: None
-    gate_lin._ensure_initialized = lambda *a, **kw: None
     gate_lin.finalize_weights()
 
     # e_bias