A small fix that makes the test suite pass.