Bug fix
Bug fixes.