diff --git a/deepposekit/models/layers/deeplabcut.py b/deepposekit/models/layers/deeplabcut.py
index 70ffd8e..2324c37 100644
--- a/deepposekit/models/layers/deeplabcut.py
+++ b/deepposekit/models/layers/deeplabcut.py
@@ -22,6 +22,7 @@
 from __future__ import print_function
 
 import os
+import numpy as np
 from tensorflow.keras.layers import Layer
 from tensorflow.keras import backend