fix a bug in the driver

Signed-off-by: Sébastien Boisvert <sebastien.boisvert.3@ulaval.ca>