Minor testing fixes
test fix