diff --git a/mediapipe/tasks/python/audio/__init__.py b/mediapipe/tasks/python/audio/__init__.py
index df217be699..e9d55254d3 100644
--- a/mediapipe/tasks/python/audio/__init__.py
+++ b/mediapipe/tasks/python/audio/__init__.py
@@ -14,9 +14,9 @@
 
 """MediaPipe Tasks Audio API."""
 
-import mediapipe.tasks.python.audio.core
-import mediapipe.tasks.python.audio.audio_classifier
-import mediapipe.tasks.python.audio.audio_embedder
+from mediapipe.tasks.python.audio import audio_classifier
+from mediapipe.tasks.python.audio import audio_embedder
+from mediapipe.tasks.python.audio import core
 
 AudioClassifier = audio_classifier.AudioClassifier
 AudioClassifierOptions = audio_classifier.AudioClassifierOptions