Updated test method snippet