Fixed check for the test