E' una funzione talmente semplice che non vale la pena complicare troppo le cose, sistemerò tutto inserendo quella condizione if che mi suggerivi.

Problema risolto, grazie mille!