diff --git a/simple_regression.py b/simple_regression.py
index f0be61e..6837942 100644
--- a/simple_regression.py
+++ b/simple_regression.py
@@ -1,6 +1,6 @@
 import tensorflow as tf
 from keras import layers
-from keras import save_model
+from keras.models import save_model
 import pandas as pd
 import numpy as np
 import matplotlib.pyplot as plt