Bug fixes.

Bug fixes.