Use the same logic here.