Fix notification removal to not rely on transitionEnd events