fix missing classes during test run