Fixed files in upgrade