Update for compatibility with latest version of Atom