Fixed test done callback declared on wrong function