diff --git a/custom/layers.py b/custom/layers.py
index 1506671..8f4b000 100644
--- a/custom/layers.py
+++ b/custom/layers.py
@@ -1,6 +1,6 @@
 import tensorflow as tf
 import math as m
-from tensorflow.python import keras
+from tensorflow import keras
 import numpy as np
 import math