Fix test.

Fix call to undefined module.