fixes a couple tests