update logic to handle test editing