From 27a0d3a93adf2d86bbea3c5092c19e3e3d50671a Mon Sep 17 00:00:00 2001
From: oow214 <oow214@gmail.com>
Date: Thu, 30 Jun 2022 21:00:10 +0900
Subject: [PATCH] Update itertools count repr

---
 vm/src/stdlib/itertools.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/vm/src/stdlib/itertools.rs b/vm/src/stdlib/itertools.rs
index 37ab88bb7..20e266b75 100644
--- a/vm/src/stdlib/itertools.rs
+++ b/vm/src/stdlib/itertools.rs
@@ -17,7 +17,7 @@ mod decl {
         AsObject, Py, PyObjectRef, PyPayload, PyRef, PyResult, PyWeakRef, VirtualMachine,
     };
     use crossbeam_utils::atomic::AtomicCell;
-    use num_traits::{Signed, ToPrimitive};
+    use num_traits::{Signed, ToPrimitive, One};
     use std::fmt;
 
     #[pyattr]
@@ -222,8 +222,8 @@ mod decl {
         #[pymethod(magic)]
         fn repr(&self, vm: &VirtualMachine) -> PyResult<String> {
             let cur = format!("{}", self.cur.read().clone().repr(vm)?);
-            let step = self.step.as_u32_mask().clone();
-            if step == 1 {
+            let step = self.step.as_bigint();
+            if step.is_one() {
                 return Ok(format!("count({})", cur));
             }
             Ok(format!("count({}, {})", cur, step.to_string()))