Fix jug import