fix buggy shitty code