diff --git a/machine_learning/local_weighted_learning/local_weighted_learning.py b/machine_learning/local_weighted_learning/local_weighted_learning.py
index 8dd0e55d41df..ada6f7cd2520 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] = item @ local_weight(item, x_train, y_train, tau)
+        y_pred[i] = np.dot(item, local_weight(item, x_train, y_train, tau))
 
     return y_pred