fixes class package for test