diff --git a/machine_learning/local_weighted_learning/local_weighted_learning.py b/machine_learning/local_weighted_learning/local_weighted_learning.py
index ada6f7cd2..f3056da40 100644
--- a/machine_learning/local_weighted_learning/local_weighted_learning.py
+++ b/machine_learning/local_weighted_learning/local_weighted_learning.py
@@ -122,7 +122,7 @@ def local_weight_regression(
     """
     y_pred = np.zeros(len(x_train))  # Initialize array of predictions
     for i, item in enumerate(x_train):
-        y_pred[i] = np.dot(item, local_weight(item, x_train, y_train, tau))
+        y_pred[i] = np.dot(item, local_weight(item, x_train, y_train, tau)).item()
 
     return y_pred