Added new assertion to tests