Removing wrong assertion, adding another one