Small fix for import paths