Fixed a test that wasn't running the correct code.