Remove deprecated API usage

:rotating_light: Depends on APIs introduced in Atom v0.153.0 :rotating_light: