Minor bug fix.

Minor bug fix.