Fixed small bug in previous commit.