Attempting to hotfix an import issue

Should probably be reviewed later...