Minor Bug fixes.
Bug fixes.