Couple Bug Fixes