diff --git a/Jenkinsfile b/Jenkinsfile
index 8340e67..bd061b7 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -20,13 +20,16 @@ pipeline {
     )
   }
 
+  environment {
+    KAGGLE_USERNAME = "${params.KAGGLE_USERNAME}"
+    KAGGLE_KEY = "${params.KAGGLE_KEY}"
+  }
+
   stages {
     stage('Run create-dataset script') {
       steps {
-        withEnv (["KAGGLE_USERNAME=${params.KAGGLE_USERNAME}", "KAGGLE_KEY=${params.KAGGLE_KEY}"]) {
           sh 'chmod +x create-dataset.py'
-          sh 'python3 ./create-dataset.py $KAGGLE_USERNAME'
-        }
+          sh 'python3 ./create-dataset.py $KAGGLE_USERNAME $KAGGLE_KEY'
       }
     }
     stage('Archive Artifacts') {
diff --git a/create-dataset.py b/create-dataset.py
index e4ebc8c..abe9649 100644
--- a/create-dataset.py
+++ b/create-dataset.py
@@ -10,8 +10,8 @@ from sklearn.model_selection import train_test_split
 
 
 def download_kaggle_dataset():
-    os.environ["KAGGLE_USERNAME"] = "vskyper"
-    os.environ["KAGGLE_KEY"] = sys.argv[1]
+    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)