diff --git a/create-dataset.py b/create-dataset.py
index 419ee4c..fac2bef 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -3,7 +3,7 @@ import os
 #from sklearn.model_selection import train_test_split
 
 
-CUTOFF = os.environ['CUTOFF']
+CUTOFF = int(os.environ['CUTOFF'])
 
 # READ DATA
 video_games = pandas.read_csv('./ium_z434686/Video_Games_Sales_as_at_22_Dec_2016.csv',