Couple of quick fixes that I broke.