Fixing minor issue during first install