Fixing a runtime error in the unit tests.