Fix unit tests

Until we move DATA initialisation into a constructor, this is the only
way to test these from unit tests.