Add extra assertion test