Fix assertions count in tests