Add code missed in previous commit.