Found a more suitable way to fix the import errors.