fixing testclasses