minor fix in tests