Didn't see you used different class