Hotfix: Add missing import