diff --git a/src_python/fl_client.py b/src_python/fl_client.py
index 231393ccb..321034682 100644
--- a/src_python/fl_client.py
+++ b/src_python/fl_client.py
@@ -89,7 +89,7 @@ def send_model(self):
 
         weights_path = self.weights_path + 'weights_' + self.graph_id + '_' + self.partition_id + ".npy"
 
-        weights = np.array(self.MODEL.get_weights())
+        weights = self.MODEL.get_weights()
 
         data = {"CLIENT_ID":self.partition_id,"WEIGHTS":weights,"NUM_EXAMPLES":self.graph_params[0]}
 
diff --git a/src_python/fl_server.py b/src_python/fl_server.py
index 498bc5a86..68c3c3755 100644
--- a/src_python/fl_server.py
+++ b/src_python/fl_server.py
@@ -115,7 +115,7 @@ def send_model(self, client_socket):
         if self.ROUNDS == self.training_cycles:
             self.stop_flag = True
 
-        weights = np.array(self.GLOBAL_WEIGHTS)
+        weights = self.GLOBAL_WEIGHTS
 
         data = {"STOP_FLAG":self.stop_flag,"WEIGHTS":weights}