Little code refactoring.
Signed-off-by: Matthieu 'Eldaryze' Chardon <matthieu@chardons.net>