[BUGFIX] Make class loading for tests work again