AM: rm unused imports
AM: rm unused imports, conditionally
AM: ASSERT(comm_same(flag)) to make sure all workers agree
AM: rm unused imports
AM: rm unused import