fix import feature importance handling