Ensure `stage not set` warning is displayed correctly