Replace assertEquals with assertEqual