Fix import against pygments 1.6