diff --git a/Chapter 03/code/traffic_prediction.py b/Chapter 03/code/traffic_prediction.py
index d8646d4..728eba3 100644
--- a/Chapter 03/code/traffic_prediction.py	
+++ b/Chapter 03/code/traffic_prediction.py	
@@ -49,7 +49,7 @@
     if item.isdigit():
         test_datapoint_encoded[i] = int(test_datapoint[i])
     else:
-        test_datapoint_encoded[i] = int(label_encoder[count].transform(test_datapoint[i]))
+        test_datapoint_encoded[i] = int(label_encoder[count].transform([test_datapoint[i]]))
         count = count + 1 
 
 test_datapoint_encoded = np.array(test_datapoint_encoded)