diff --git a/MOAFTest.py b/MOAFTest.py
index a0761fa..bbb6c00 100644
--- a/MOAFTest.py
+++ b/MOAFTest.py
@@ -73,7 +73,7 @@ def main():
         fusion_depth_list = [int(ch) for ch in model_type[2:]]
         model = MOAFWithSE(fusion_depth_list).to(device)
     elif "mmlp" in model_type:
-        model = MOAFWithMMLP(fusion_depth_list).to(device)
+        model = MOAFWithMMLP().to(device)
     else:
         model = MOAFNoFusion().to(device)