[FIX] Tests were failing because some idiot (me) forgot to assign the context for the tests after the refactor. D'oh