Fixed some figure code