Il confronto lo devi fare sul campo id di $return?
Allora l'if deve essere

if ( $i > $return[$i]['id'] )