Fixes an issue with an import not existing