diff --git a/audio/audiotools/post.py b/audio/audiotools/post.py
index 9272e324f..1e08b13ff 100644
--- a/audio/audiotools/post.py
+++ b/audio/audiotools/post.py
@@ -6,15 +6,14 @@ import tempfile
 import typing
 import zipfile
 from pathlib import Path
-from typing import TYPE_CHECKING
 
-if TYPE_CHECKING:
-    from audio.audiotools import AudioSignal
 import markdown2 as md
 import matplotlib.pyplot as plt
 import paddle
 from IPython.display import HTML
 
+from audio.audiotools.core import AudioSignal
+
 
 def audio_table(
         audio_dict: dict,