Fixed test not cleaning up properly