Fix leaky test