Remove usage of deprecated APIs

:rotating_light: Don't merge until Atom v0.153.0 is released :rotating_light: