Fixed bug

Signed-off-by: Mohamad Gebai <mohamad.gebai@polymtl.ca>