diff --git a/create-dataset.py b/create-dataset.py
index 227ed49..90249ae 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -1,4 +1,5 @@
 import os
+import sys
 from kaggle.api.kaggle_api_extended import KaggleApi
 import pandas as pd
 import numpy as np
@@ -10,6 +11,8 @@ from sklearn.model_selection import train_test_split
 
 def download_kaggle_dataset():
     print(f'Kaggle API key: {os.environ["KAGGLE_KEY"]}')
+    os.environ["KAGGLE_USERNAME"] = sys.argv[1]
+    os.environ["KAGGLE_KEY"] = sys.argv[2]
     kaggle = KaggleApi()
     kaggle.authenticate()
     kaggle.dataset_download_files("mlg-ulb/creditcardfraud", path="./", unzip=True)