fix condition for printing @return