Fix bug due to renaming of return value.