Fixes module build
Fixes module build