Fix defect in the code

The tests were failing and I don't like it when the tests
are failing.