Fixing minor bug.
Fixing minor bug.