Fix small bug in previous commit.