diff --git a/homeworks/assignment0_01_knn/k_nearest_neighbor.py b/homeworks/assignment0_01_knn/k_nearest_neighbor.py
index 57cc0c6ef..20d74a879 100644
--- a/homeworks/assignment0_01_knn/k_nearest_neighbor.py
+++ b/homeworks/assignment0_01_knn/k_nearest_neighbor.py
@@ -75,7 +75,7 @@ def compute_distances_two_loops(self, X):
                 # not use a loop over dimension, nor use np.linalg.norm().          #
                 #####################################################################
                 # *****START OF YOUR CODE (DO NOT DELETE/MODIFY THIS LINE)*****
-
+                dists = np.sqrt(np.sum(np.square(X[i] - self.X_train[j]), axis=1))
                 # *****END OF YOUR CODE (DO NOT DELETE/MODIFY THIS LINE)*****
         return dists