Small refactoring

Tu devrais tester un peu avant :)