diff --git a/elements/main.py b/elements/main.py
index ddc33e25797f36ec6f4574055e65a34969be05c7..60722db37732ad6d5b35865efea4f5906e1fc57b 100644
--- a/elements/main.py
+++ b/elements/main.py
@@ -50,9 +50,9 @@ class BeamSplitter(Element):
         self.output_vector = self.matrix @ input_state
 
         if side == "left":
-            ans, _ = np.split(self.output_vector)
+            ans, _ = np.split(self.output_vector, 2)
 
         elif side == "right":
-            _, ans = np.split(self.output_vector)
+            _, ans = np.split(self.output_vector, 2)
 
         return ans