diff --git a/move/switchboard/sources/schemas/aggregator.move b/move/switchboard/sources/schemas/aggregator.move
index f2c7f90..8b1d8d1 100644
--- a/move/switchboard/sources/schemas/aggregator.move
+++ b/move/switchboard/sources/schemas/aggregator.move
@@ -1,5 +1,5 @@
 module switchboard::aggregator {
-    use aptos_framework::account::{Self, SignerCapability};
+    use aptos_framework::account::{SignerCapability};
     use aptos_framework::timestamp;
     use aptos_framework::block;
     use aptos_std::ed25519;
@@ -562,6 +562,9 @@ module switchboard::aggregator {
         }
     }
 
+    #[test_only]
+    use aptos_framework::account;
+
     #[test_only]
     public entry fun new_test(account: &signer, value: u128, dec: u8, neg: bool) {
         let cap = account::create_test_signer_cap(signer::address_of(account));