Revert "Make code match doc"

This reverts commit 6cb48b5eb7f3ddf7da0220bb4c67a6dcb9dfa89c.