Fix test snapshotting