Fix small bug after previous commit.