Add tests for helper lookup