Fix installation issue