Use assert instead of should