Update: remove inconsistencies in tests and docstring