minor fix for tests