Fix up getDirection, and imports in 2 TE classes

To make it compile with newer Forge versions