diff --git a/create-dataset.py b/create-dataset.py
index d9b8adb..e48efe2 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -1,5 +1,5 @@
 import os
-import kaggle
+from kaggle.api.kaggle_api_extended import KaggleApi
 import pandas as pd
 import numpy as np
 
@@ -9,8 +9,9 @@ from sklearn.model_selection import train_test_split
 
 
 def download_kaggle_dataset():
-    kaggle.api.authenticate()
-    kaggle.api.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)
+    kaggle = KaggleApi()
+    kaggle.authenticate()
+    kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)
 
 
 def load_data(name):