Fix wrong function