Refactoring

J'ai pas testé mais ça devrai marcher :+1: