How about an updated test for this change?