Minor update to the documentation
Fixed according to the new module API