Switch to pm-runtime instead