Moving test check to init method