fix some stuff I broke