Fix wrong refactoring.

Signed-off-by: Hugues Peccatte <hugues.peccatte@gmail.com>