Clean testing code