Adding test helper functions