И поведением тоже надо заниматься. Поговорите с учителем, выработайте совместный комплекс мер. Моя иногда выключается на уроках т.е. задумывается и уходит в себя. Я предупредила учтительницу заранее о том что ребенок способен выключаться и она научилась включать ребенка, даже как то показала мне условный жест, которым привлекает ее внимание т.е. не отчитывает перед классом, а делает это незаметно для остальных. И к концу первого класса отметила что таких выключений стало значительно меньше т.е. ребенок стал способен удерживать внимание.Ваш, видимо по той же причине бегает, если учитель пойдет навстречу и найдет способ как ребенок может  снять нервное напряжение, не мешая остальным, то тоже все наладится со временем.