fixed bug (i think)