Fix figure again