also fix the docs