Petite simplification

Petite simplification