I copied this from another test... I think I should have `import unittest` here too right? Probably need to check other unit test files for this also.