Fixed small bug in last commit.