Don't you need to print the unit in the else branch?