Fix a bug in my previous commit.