Updated test method