Simple fix for several broken tests

*boom*