diff --git a/mypy_nonfloat_decimal/mypy_nonfloat_decimal.py b/mypy_nonfloat_decimal/mypy_nonfloat_decimal.py index 8fedb3a..dc3ee0b 100644 --- a/mypy_nonfloat_decimal/mypy_nonfloat_decimal.py +++ b/mypy_nonfloat_decimal/mypy_nonfloat_decimal.py @@ -29,7 +29,7 @@ def get_function_hook( :param fullname: full name of the type to consider :return: callback for specific type call """ - if fullname == "decimal.Decimal": + if fullname in ("_decimal.Decimal", "decimal.Decimal"): return analyze_decimal_call return None