Fix documentation after module re-organization