Replace 'should equal' with 'assertEqual' to avoid type-safety problems with the tests.