Fixing assert to work with multiple args