Fixing bug

Signed-off-by: Kevin Nguyen <k.nguyen@thecodingmachine.com>