fix import in rootpy script