Fix failing test. All tests now pass.