Changed assertTrue for assertEqual