Change assert to assertTrue