fix tests
Merge pull request #304 from ministryofjustice/fix-tests

fix tests