Fix a bug

Signed-off-by: Damien Cassou <damien.cassou@gmail.com>