Fixes module reference
Fixes return value