minor fix to tests