Hotfix for missing import