small code simplication