Change code to fix a bug made in previous commit.