Update to work with github's latest UI changes