Fix jsdoc after module refactor