Whoops, forgot to fix a test.