assert statement fix