Fixes broken tests.
The test runner faked me out. =\