Fix failing test file

The test file use'd the old name of the module. This caused the test to fail.