Adding test for balance provider