diff --git a/keras_cv/src/layers/vit_layers.py b/keras_cv/src/layers/vit_layers.py
index 5b0a4c4d3d..b9693a27d1 100644
--- a/keras_cv/src/layers/vit_layers.py
+++ b/keras_cv/src/layers/vit_layers.py
@@ -15,10 +15,10 @@
 import math
 
 import tensorflow as tf
-from keras_cv.backend import layers
-from keras_cv.backend import ops
+from keras_cv.src.backend import layers
+from keras_cv.src.backend import ops
 
-from keras_cv.api_export import keras_cv_export
+from keras_cv.src.api_export import keras_cv_export
 
 
 @keras_cv_export("keras_cv.layers.PatchingAndEmbedding")