snapshot: wait for snapshot to complete

Change-Id: I13a70e5df281ad08bc148f1d3f29f6b893ef6e1a