Fixes test module imports.