diff --git a/neuralogic/core/constructs/java_objects.py b/neuralogic/core/constructs/java_objects.py
index c7fc8d94..677b441a 100644
--- a/neuralogic/core/constructs/java_objects.py
+++ b/neuralogic/core/constructs/java_objects.py
@@ -36,7 +36,7 @@ def get_value(self, weight):
                     value = self.vector_value(weight[0])
             elif len(weight) == 2:
                 if weight[0] == 1:
-                    value = self.vector_value(weight[1], True)
+                    value = self.scalar_value() if weight[1] == 1 else self.vector_value(weight[1], True)
                 elif weight[1] == 1:
                     value = self.vector_value(weight[0], False)
                 else: