Fix plot title

AnkitBarik · Apr 21, 2021 · cb52a78 · cb52a78
1 parent fc3ddbf
commit cb52a78
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/planetmagfields/planet.py b/planetmagfields/planet.py
@@ -52,14 +52,12 @@ def plot(self,r=1,levels=30,cmap='RdBu_r',proj='Mollweide'):
         if proj.lower() != 'hammer' and self.name == 'earth':
             ax.coastlines()
 
-        ax.set_title(planet.capitalize() + radLabel,fontsize=25,pad=20)
-
         if r==1:
             radLabel = '  Surface'
         else:
             radLabel = r'  $r/r_{\rm surface}=%.2f$' %r
 
-        ax.set_title(planet.capitalize() + radLabel,fontsize=25,pad=20)
+        ax.set_title(self.name.capitalize() + radLabel,fontsize=25,pad=20)
         plt.tight_layout()
 
     def writeVtsFile(self,potExtra=False,ratio_out=2,nrout=32):