Fixing a couple broken tests