Fix the annoying PM bug
Thanks to Diego Andrés for the fix

Signed-off-by: Suki <suki@missallsunday.com>