diff --git a/data_prepare.py b/data_prepare.py
index 5c44f82..1e8cb75 100644
--- a/data_prepare.py
+++ b/data_prepare.py
@@ -3,8 +3,8 @@ import os
 
 # Set Java environment variable
 os.environ['JAVA_HOME'] = '/opt/module/jdk1.8.0_171'
-os.environ["PYSPARK_PYTHON"]="/usr/local/bin/python3.6"
-os.environ["PYSPARK_DRIVER_PYTHON"]="/usr/local/bin/python3.6"
+os.environ["PYSPARK_PYTHON"]="/usr/bin/python3"
+os.environ["PYSPARK_DRIVER_PYTHON"]="/usr/bin/python3"
 
 # Parse ratings data into (userID, movieID, rating)
 def get_ratings_tuple(entry):