add back missing imports