This test wasn't using decorators before and was missing the import, so my previous commit broke the test.