Fixed issue with Kaggle script.