[BUGFIX] Mention correct class and function in rst file