Correct bug on install