Fix a small bug with the last commit.