diff --git a/06_Neural_Networks/07_Improving_Linear_Regression/07_improving_linear_regression.py b/06_Neural_Networks/07_Improving_Linear_Regression/07_improving_linear_regression.py
index b81838d26..d7deb3d41 100644
--- a/06_Neural_Networks/07_Improving_Linear_Regression/07_improving_linear_regression.py
+++ b/06_Neural_Networks/07_Improving_Linear_Regression/07_improving_linear_regression.py
@@ -64,7 +64,7 @@
 
 # Split data into train/test = 80%/20%
 train_indices = np.random.choice(len(x_vals), round(len(x_vals)*0.8), replace=False)
-test_indices = np.array(list(set(range(len(x_vals))) - set(train_indices)))
+test_indices = list(set(range(len(x_vals))) - set(train_indices))
 x_vals_train = x_vals[train_indices]
 x_vals_test = x_vals[test_indices]
 y_vals_train = y_vals[train_indices]
diff --git a/test.txt b/test.txt
new file mode 100644
index 000000000..e69de29bb