fix damn bug!