think i might have fixed the bug