Fix Flavien/Cedric TODO - Adrien
Fixed one more todo - adrien