fixed bug, i think