Remove usage of deprecated APIs

:rotating_light: Depends on 0.153.0 :rotating_light: