Cleans up testing code