diff --git a/nnfwtbn/model.py b/nnfwtbn/model.py
index e9ede9a1172a9bcd63db62b15ae54aa6c8d1f537..98982a6b248ed2e1626082fe89656cf00de6e462 100644
--- a/nnfwtbn/model.py
+++ b/nnfwtbn/model.py
@@ -7,7 +7,7 @@ import json
 
 import numpy as np
 import pandas as pd
-import tensorflow.keras as keras
+import keras
 
 from nnfwtbn.variable import Variable
 from nnfwtbn.helpers import python_to_str, str_to_python