diff --git a/examples/frameworks/keras/manual_model_upload.py b/examples/frameworks/keras/manual_model_upload.py
index 11c8e6de..7aec6518 100644
--- a/examples/frameworks/keras/manual_model_upload.py
+++ b/examples/frameworks/keras/manual_model_upload.py
@@ -3,7 +3,7 @@
 import os
 from tempfile import gettempdir
 
-from keras import Input, layers, Model
+from tensorflow.keras import Input, layers, Model
 
 from clearml import Task