Import the main module from the correct place
Merge pull request #1 from alphagov/fix-tests

Import the main module from the correct place