Fixes an import error