Post install script fix