Use assertTrue instead of plain assert