Couple Bug fixes