Add check to test