Fix leak in UTs