diff --git a/compiler/modules/write_mask_and_array.py b/compiler/modules/write_mask_and_array.py
index c949fe58..9c077ed7 100644
--- a/compiler/modules/write_mask_and_array.py
+++ b/compiler/modules/write_mask_and_array.py
@@ -28,7 +28,7 @@ class write_mask_and_array(design.design):
         self.columns = columns
         self.word_size = word_size
         self.write_size = write_size
-        self.column_offset
+        self.column_offset = column_offset
         self.words_per_row = int(columns / word_size)
         self.num_wmasks = int(word_size / write_size)
 
diff --git a/compiler/tests/07_single_level_column_mux_array_test.py b/compiler/tests/07_single_level_column_mux_array_test.py
index ff7ec1c9..b1f74eba 100755
--- a/compiler/tests/07_single_level_column_mux_array_test.py
+++ b/compiler/tests/07_single_level_column_mux_array_test.py
@@ -21,15 +21,15 @@ class single_level_column_mux_test(openram_test):
         globals.init_openram(config_file)
 
         debug.info(1, "Testing sample for 2-way column_mux_array")
-        a = factory.create(module_type="single_level_column_mux_array", columns=16, port=0, word_size=8)
+        a = factory.create(module_type="single_level_column_mux_array", columns=16, word_size=8)
         self.local_check(a)
 
         debug.info(1, "Testing sample for 4-way column_mux_array")
-        a = factory.create(module_type="single_level_column_mux_array", columns=16, port=0, word_size=4)
+        a = factory.create(module_type="single_level_column_mux_array", columns=16, word_size=4)
         self.local_check(a)
 
         debug.info(1, "Testing sample for 8-way column_mux_array")
-        a = factory.create(module_type="single_level_column_mux_array", columns=32, port=0, word_size=4)
+        a = factory.create(module_type="single_level_column_mux_array", columns=32, word_size=4)
         self.local_check(a)
 
         globals.end_openram()