Fix in the wrong function