Clean up model tests