think i fixed the bug