Installation script fix