Fix the build, forgot to remove module