Fix test for first method