Fix test assertions