diff --git a/Chapter 02/code/income_classifier.py b/Chapter 02/code/income_classifier.py
index 740095f..ed13549 100644
--- a/Chapter 02/code/income_classifier.py	
+++ b/Chapter 02/code/income_classifier.py	
@@ -75,12 +75,12 @@
     if item.isdigit():
         input_data_encoded[i] = int(input_data[i])
     else:
-        input_data_encoded[i] = int(label_encoder[count].transform(input_data[i]))
+        input_data_encoded[i] = int(label_encoder[count].transform([input_data[i]]))
         count += 1 
 
 input_data_encoded = np.array(input_data_encoded)
 
 # Run classifier on encoded datapoint and print output
-predicted_class = classifier.predict(input_data_encoded)
+predicted_class = classifier.predict([input_data_encoded])
 print(label_encoder[-1].inverse_transform(predicted_class)[0])