Use assertTrue instead of assertEqual