Add test helper methods