diff --git a/contracts/integrations/test/internal-integration-tests/match_orders_test.ts b/contracts/integrations/test/internal-integration-tests/match_orders_test.ts
index 027dab6c88..f95771a722 100644
--- a/contracts/integrations/test/internal-integration-tests/match_orders_test.ts
+++ b/contracts/integrations/test/internal-integration-tests/match_orders_test.ts
@@ -1,8 +1,7 @@
 import { DevUtilsContract } from '@0x/contracts-dev-utils';
 import { BlockchainBalanceStore, TokenIds } from '@0x/contracts-exchange';
 import { ReferenceFunctions as LibReferenceFunctions } from '@0x/contracts-exchange-libs';
-import { toBaseUnitAmount } from '@0x/contracts-staking';
-import { blockchainTests, constants, expect } from '@0x/contracts-test-utils';
+import { blockchainTests, constants, expect, toBaseUnitAmount } from '@0x/contracts-test-utils';
 import { assetDataUtils, ExchangeRevertErrors, orderHashUtils } from '@0x/order-utils';
 import { Order, OrderStatus, SignedOrder } from '@0x/types';
 import { BigNumber } from '@0x/utils';
diff --git a/contracts/test-utils/src/number_utils.ts b/contracts/test-utils/src/number_utils.ts
index ead259022d..3fc0087b55 100644
--- a/contracts/test-utils/src/number_utils.ts
+++ b/contracts/test-utils/src/number_utils.ts
@@ -81,10 +81,10 @@ export function assertIntegerRoughlyEquals(actual: Numberish, expected: Numberis
 /**
  * Converts `amount` into a base unit amount with 18 digits.
  */
-export function toBaseUnitAmount(amount: Numberish): BigNumber {
-    const decimals = 18;
+export function toBaseUnitAmount(amount: Numberish, decimals?: number): BigNumber {
+    const baseDecimals = decimals !== undefined ? decimals : 18;
     const amountAsBigNumber = new BigNumber(amount);
-    const baseUnitAmount = Web3Wrapper.toBaseUnitAmount(amountAsBigNumber, decimals);
+    const baseUnitAmount = Web3Wrapper.toBaseUnitAmount(amountAsBigNumber, baseDecimals);
     return baseUnitAmount;
 }