small fix on the test method