Fix previous code

Signed-off-by: Bruno Lesieur <bruno.lesieur@gmail.com>