minor fix on the module