Added test helper functions