forgot to fix getting random shape.