Fix: bug

Signed-off-by: Gregory Detal <gregory.detal@uclouvain.be>