diff --git a/script5_1.py b/script5_1.py
index 67528c0..d62c05a 100644
--- a/script5_1.py
+++ b/script5_1.py
@@ -13,7 +13,8 @@ data = pd.read_csv('data.csv')
 
 # Prepare the data
 X = data[['movie title', 'User Rating', 'Director', 'Top 5 Casts', 'Writer']]
-y = data['Rating'].values.astype(float)
+y = pd.Series(data['Rating'], dtype=float)
+# y = data['Rating'].values.astype(float)
 
 print("Data type of 'Rating' column:", y.dtype)