O código do @vitornp está melhor, não precisa do if.