Add lost code due to reload from repo