Fixing a little incoherence of the code