Update test assert