fixing stuff I broke in last commit