diff --git a/compiler/tests/10_write_mask_and_array_test.py b/compiler/tests/10_write_mask_and_array_test.py
index 91155467..6334ac97 100755
--- a/compiler/tests/10_write_mask_and_array_test.py
+++ b/compiler/tests/10_write_mask_and_array_test.py
@@ -24,15 +24,15 @@ class write_mask_and_array_test(openram_test):
 
         # check write driver array for single port
         debug.info(2, "Testing write_mask_and_array for columns=8, word_size=8, write_size=4")
-        a = factory.create(module_type="write_mask_and_array", columns=8, word_size=8, write_size=4)
+        a = factory.create(module_type="write_mask_and_array", columns=8, word_size=8, write_size=4, port=0)
         self.local_check(a)
 
         debug.info(2, "Testing write_mask_and_array for columns=16, word_size=16, write_size=4")
-        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=16, write_size=4)
+        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=16, write_size=4, port=0)
         self.local_check(a)
 
         debug.info(2, "Testing write_mask_and_array for columns=16, word_size=8, write_size=2")
-        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=8, write_size=2)
+        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=8, write_size=2, port=0)
         self.local_check(a)
 
         # check write driver array for multi-port
@@ -43,11 +43,11 @@ class write_mask_and_array_test(openram_test):
 
         factory.reset()
         debug.info(2, "Testing write_mask_and_array for columns=8, word_size=8, write_size=4 (multi-port case)")
-        a = factory.create(module_type="write_mask_and_array", columns=8, word_size=8, write_size=4)
+        a = factory.create(module_type="write_mask_and_array", columns=8, word_size=8, write_size=4, port=0)
         self.local_check(a)
 
         debug.info(2, "Testing write_mask_and_array for columns=16, word_size=8, write_size=2 (multi-port case)")
-        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=8, write_size=2)
+        a = factory.create(module_type="write_mask_and_array", columns=16, word_size=8, write_size=2, port=0)
         self.local_check(a)
 
         globals.end_openram()