Fix check for test