Use assert in tests