fix tests after module name change

Forgot to include this in the last commit.