Added test assertions