Fix that multiple screenshots of one step are taken before the step is closed. Bug: 321157648 Test: locally Change-Id: Ia1430765885bfbb0b4e31a5fc34a8c82112d7185