diff --git a/beetsplug/chroma.py b/beetsplug/chroma.py
index 8259934fd..f90877113 100644
--- a/beetsplug/chroma.py
+++ b/beetsplug/chroma.py
@@ -25,7 +25,8 @@ import acoustid
 import confuse
 
 from beets import config, ui, util
-from beets.autotag import Distance, TrackInfo
+from beets.autotag.distance import Distance
+from beets.autotag.hooks import TrackInfo
 from beets.metadata_plugins import MetadataSourcePlugin
 from beetsplug.musicbrainz import MusicBrainzPlugin