Fixed error on figure input