Fix small bug in last commit.