diff --git a/JenkinsfileNeural b/JenkinsfileNeural
index 02c8809..b42fa9c 100644
--- a/JenkinsfileNeural
+++ b/JenkinsfileNeural
@@ -12,7 +12,7 @@ node {
 
         }
         stage('Clone repo') {
-                    docker.image("karopa/ium:09").inside {
+                    docker.image("karopa/ium:10").inside {
                         stage('Test') {
                             checkout([$class: 'GitSCM', branches: [[name: '*/master']], doGenerateSubmoduleConfigurations: false, extensions: [], submoduleCfg: [], userRemoteConfigs: [[url: 'https://git.wmi.amu.edu.pl/s434765/ium_434765']]])
                             copyArtifacts fingerprintArtifacts: true, projectName: 's434765-create-dataset', selector: buildParameter("BUILD_SELECTOR")
diff --git a/neural_network.py b/neural_network.py
index e193edd..07de086 100644
--- a/neural_network.py
+++ b/neural_network.py
@@ -39,7 +39,7 @@ model.compile(loss='mean_absolute_error', optimizer="Adam", metrics=['mean_absol
 
 model.fit(X, y, epochs=30, validation_split = 0.3)
 
-data = pd.read_csv("data_dev", sep=',', error_bad_lines=False, skip_blank_lines=True).dropna()
+data = pd.read_csv("data_dev", sep=',', error_bad_lines=False, skip_blank_lines=True, nrows=527).dropna()
 X_test = data.loc[:,data.columns == "440265"].astype(int)
 y_test = data.loc[:,data.columns == "21629"].astype(int)