Fix hardcoded module names which cause issues with the 'unicode_literals' versions of the tests