Fix count Bug

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