fixed bug, i think?, I hope