Fix LTIModule imports