Made tiny changes to MarkTest RestoreTest