Writing code to fix a test. :(