Not sure if this was correct, but it was required to make the test pass.