From 6b28cf7bb73138e781d06ba9f8058a9087139f03 Mon Sep 17 00:00:00 2001
From: Douglas Wilson <douglas.wilson@quantinuum.com>
Date: Fri, 2 Feb 2024 10:51:31 +0000
Subject: [PATCH] Improve test coverage

---
 src/std_extensions/arithmetic/int_types.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/std_extensions/arithmetic/int_types.rs b/src/std_extensions/arithmetic/int_types.rs
index 53f123657f..ffd9e5f595 100644
--- a/src/std_extensions/arithmetic/int_types.rs
+++ b/src/std_extensions/arithmetic/int_types.rs
@@ -150,7 +150,7 @@ impl CustomConst for ConstIntU {
         format!("u{}({})", self.log_width, self.value).into()
     }
     fn check_custom_type(&self, typ: &CustomType) -> Result<(), CustomCheckFailure> {
-        if typ == &int_custom_type(type_arg(self.log_width)) {
+        if typ == &self.typ() {
             Ok(())
         } else {
             Err(CustomCheckFailure::Message(
@@ -179,7 +179,7 @@ impl CustomConst for ConstIntS {
         format!("i{}({})", self.log_width, self.value).into()
     }
     fn check_custom_type(&self, typ: &CustomType) -> Result<(), CustomCheckFailure> {
-        if typ == &int_custom_type(type_arg(self.log_width)) {
+        if typ == &self.typ() {
             Ok(())
         } else {
             Err(CustomCheckFailure::Message(