Fix assertEquals error