fixed bug, I think