removed class in tests