Add helper functions for mocking