Fixed tests by referring to new module name. Issue: #415