Fix in key_path statements to match updated method usage