Making the small record test case slightly better