Fixing only/last-child problem