Modified test function.