Update assertion

Making things better. :smile:
Looks like the wrong assertion is happening here. Let's fix it.