pass in wait params for snapshot

Change-Id: I03eceb1174bc6857e6bcc95debc479772b61d8ff