fix the test due the change on #6