Check correct class