Fixes some tests that weren't updated like they should have been.

I'm still getting used to testing ...