A hotfix for importing the classes