re-balancing mocking logic