Fixes module reference in test.