diff --git a/crates/pink/runtime/src/runtime/extension.rs b/crates/pink/runtime/src/runtime/extension.rs
index cc1fde83e..8953a104c 100644
--- a/crates/pink/runtime/src/runtime/extension.rs
+++ b/crates/pink/runtime/src/runtime/extension.rs
@@ -267,7 +267,6 @@ impl PinkExtBackend for CallInQuery {
         &self,
         account: ext::AccountId,
     ) -> Result<(pink::Balance, pink::Balance), Self::Error> {
-        self.ensure_system()?;
         let account: AccountId32 = account.convert_to();
         let total = crate::runtime::Balances::total_balance(&account);
         let free = crate::runtime::Balances::free_balance(&account);
diff --git a/crates/pink/runtime/tests/test_pink_contract.rs b/crates/pink/runtime/tests/test_pink_contract.rs
index 0477e68e9..5395f36a2 100644
--- a/crates/pink/runtime/tests/test_pink_contract.rs
+++ b/crates/pink/runtime/tests/test_pink_contract.rs
@@ -280,7 +280,7 @@ fn test_get_balance() {
         .call()
         .direct_balance_of(TEST_ADDRESS.convert_to())
         .query(&mut cluster);
-    assert!(result.is_err());
+    assert_eq!(result.unwrap().0, balance);
 
     let result = checker
         .call()