Change code to conform to documentation.