Fixed broken tests. Oops.