Update LifecycleCallbackTest with new return type