Fix shape type.

Change-Id: Iad25c64ed6ed491e3b9448f3f0966a82152140d9