fixed an assertion error in the test