Improve assert snippet