add assertion into test