diff --git a/machine_learning/random_forest_regression/random_forest_regression.py b/machine_learning/random_forest_regression/random_forest_regression.py
index b984b02b6e6e..85ce0676b598 100644
--- a/machine_learning/random_forest_regression/random_forest_regression.py
+++ b/machine_learning/random_forest_regression/random_forest_regression.py
@@ -30,7 +30,7 @@
 regressor.fit(X, y)
 
 # Predicting a new result
-y_pred = regressor.predict(6.5).reshape(1, -1)
+y_pred = regressor.predict([[6.5]])
 
 # Visualising the Random Forest Regression results (higher resolution)
 X_grid = np.arange(min(X), max(X), 0.01)