Fixes another import error