diff --git a/training.jenkinsfile b/training.jenkinsfile
index a60173f..7eb3f0d 100644
--- a/training.jenkinsfile
+++ b/training.jenkinsfile
@@ -24,7 +24,7 @@ node {
 
 	local_image.inside {
 		stage('Build') {
-			checkout([$class: 'GitSCM', branches: [[name: 'ztm']], extensions: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s452639/ium_452639']]])
+			checkout([$class: 'GitSCM', branches: [[name: BRANCH_NAME]], extensions: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s452639/ium_452639']]])
 
 			copyArtifacts fingerprintArtifacts: true,
 				projectName: 's452639-create-dataset',