Fix misplaced code.