Fix attribute importance finder