update docstring with correct type info