I thought about that phrasing, but it was too far from the original quote for me. But yes, it also works and is the better joke as well.