last minute bug fix yay