fix includes in a couple of tests