diff --git a/main.py b/main.py
index 5b43723..aaed02a 100755
--- a/main.py
+++ b/main.py
@@ -50,6 +50,6 @@ if __name__ == '__main__':
 
     to_serialise = qam.demodulate(to_decode)
 
-    data = serialise(to_serialise)
+    data = serialise(16, to_serialise)
 
     print(data)
diff --git a/qam.py b/qam.py
index a052442..909ff80 100644
--- a/qam.py
+++ b/qam.py
@@ -34,7 +34,7 @@ def demodulate(in_data):
     out_data = np.ndarray((len(in_data), len(in_data[0])), dtype=np.uint8)
 
     # Just pull the constellation array data out
-    constellation = { x for x in qam_demapping_table.keys() }
+    constellation = [ x for x in qam_demapping_table.keys() ]
 
     for i in range(len(in_data)):
         for j in range(len(in_data[0])):