fixed bug on previous commit (woops)