fix a couple tests