diff --git a/tmva/pymva/test/testPyKerasClassification.C b/tmva/pymva/test/testPyKerasClassification.C
index 8e71e29eafe4605931d2c7d51611a0076a69e722..67b2a18393553ba81e5678b8686298620daa6dcf 100644
--- a/tmva/pymva/test/testPyKerasClassification.C
+++ b/tmva/pymva/test/testPyKerasClassification.C
@@ -12,12 +12,11 @@
 TString pythonSrc = "\
 from keras.models import Sequential\n\
 from keras.layers.core import Dense, Activation\n\
-from keras import initializations\n\
 from keras.optimizers import SGD\n\
 \n\
 model = Sequential()\n\
-model.add(Dense(64, init=\"normal\", activation=\"relu\", input_dim=4))\n\
-model.add(Dense(2, init=\"normal\", activation=\"softmax\"))\n\
+model.add(Dense(64, activation=\"relu\", input_dim=4))\n\
+model.add(Dense(2, activation=\"softmax\"))\n\
 model.compile(loss=\"categorical_crossentropy\", optimizer=SGD(lr=0.01), metrics=[\"accuracy\",])\n\
 model.save(\"kerasModelClassification.h5\")\n";
 
diff --git a/tmva/pymva/test/testPyKerasMulticlass.C b/tmva/pymva/test/testPyKerasMulticlass.C
index ddf88bd16abffa1124adc046d02aae6a87543453..b62272732b755f3fe05350b310642a068f7f6c71 100644
--- a/tmva/pymva/test/testPyKerasMulticlass.C
+++ b/tmva/pymva/test/testPyKerasMulticlass.C
@@ -13,12 +13,11 @@
 TString pythonSrc = "\
 from keras.models import Sequential\n\
 from keras.layers.core import Dense, Activation\n\
-from keras import initializations\n\
 from keras.optimizers import Adam\n\
 \n\
 model = Sequential()\n\
-model.add(Dense(64, init=\"glorot_normal\", activation=\"relu\", input_dim=4))\n\
-model.add(Dense(4, init=\"glorot_normal\", activation=\"softmax\"))\n\
+model.add(Dense(64, activation=\"relu\", input_dim=4))\n\
+model.add(Dense(4, activation=\"softmax\"))\n\
 model.compile(loss=\"categorical_crossentropy\", optimizer=Adam(), metrics=[\"accuracy\",])\n\
 model.save(\"kerasModelMulticlass.h5\")\n";
 
diff --git a/tmva/pymva/test/testPyKerasRegression.C b/tmva/pymva/test/testPyKerasRegression.C
index 4313ceba13754a26c054bb166d80efc3a066c167..030ebb87b90d3f97cad7bd0e137086bc6659db5e 100644
--- a/tmva/pymva/test/testPyKerasRegression.C
+++ b/tmva/pymva/test/testPyKerasRegression.C
@@ -12,12 +12,11 @@
 TString pythonSrc = "\
 from keras.models import Sequential\n\
 from keras.layers.core import Dense, Activation\n\
-from keras import initializations\n\
 from keras.optimizers import SGD\n\
 \n\
 model = Sequential()\n\
-model.add(Dense(64, init=\"normal\", activation=\"tanh\", input_dim=2))\n\
-model.add(Dense(1, init=\"normal\", activation=\"linear\"))\n\
+model.add(Dense(64, activation=\"tanh\", input_dim=2))\n\
+model.add(Dense(1, activation=\"linear\"))\n\
 model.compile(loss=\"mean_squared_error\", optimizer=SGD(lr=0.01))\n\
 model.save(\"kerasModelRegression.h5\")\n";