minor fix to test