Fixed issue #2.

Signed-off-by: Pier-Luc Plante <pier-luc.plante.1@ulaval.ca>