Petit refactoring et bugfix