Update assertion to fix failing test