Docs correction
Update tests