Fix misplaced include