Fixes code to match the comments.