Remove un-needed assertion.
Update docstring.