Fixed figure build error