Fix tests for snapshot tasks