improve test assertions