Bug fix for install process