:green_heart: Fix broken tests

Not sure why but I have these tests failing on local.