minor fix on test