think i fixed bug