Bug fix!!~

Something I changed here caused a unit test to fail. That's no longer a
problem.