Fix invalid import left over after issue 4 fix.