Fixed a bug.

Fixed a bug.