fixing broken tests. oops.