minor fix for test