updated class name in tests