Cleanup after tests

Was causing an error with later tests