Update assertion

Making things better. :smile: