Added test assertion