bug fix lololl