diff --git a/{{cookiecutter.project_name}}/Keras_Tensorflow/environment.yml b/{{cookiecutter.project_name}}/Keras_Tensorflow/environment.yml
index 7521eb6..62064de 100644
--- a/{{cookiecutter.project_name}}/Keras_Tensorflow/environment.yml
+++ b/{{cookiecutter.project_name}}/Keras_Tensorflow/environment.yml
@@ -21,7 +21,7 @@ dependencies:
   - azure-cli==2.0.63
   - azure-core
   - keras==2.2.0
-  - azureml-sdk[notebooks, contrib]==1.0.57
+  - azureml-core==1.0.57
   - locustio==0.11.0
   - prompt-toolkit==2.0.9
   - git+https://github.com/microsoft/AI-Utilities.git