bug fix lol