Install script fix