diff --git a/codes/model.py b/codes/model.py
index cad2a5bdd2d4a2dcf0e8aeee1481b699a95d2937..9606a2e69ac5c844f9fc57df6d8aca47bfa73a5a 100644
--- a/codes/model.py
+++ b/codes/model.py
@@ -34,14 +34,13 @@ class Model:
     def calculateEF(self):
         self.EF = fermiOnGrid(self.kham, self.filling)
 
-    def makeDensityMatrix(self, mf_model, nK=200):
+    def makeDensityMatrixTb(self, mf_model, nK=200):
         self.kham = tb2kham(addTb(self.h_0, mf_model), nK=nK, ndim=self._ndim)
         self.calculateEF()
-        return densityMatrix(self.kham, self.EF)
+        return ifftn2tb(kdens2tbFFT(densityMatrix(self.kham, self.EF), self._ndim))
 
     def mfield(self, mf_model, nK=200):
-        densityMatrix = self.makeDensityMatrix(mf_model, nK=nK)
-        densityMatrixTb = ifftn2tb(kdens2tbFFT(densityMatrix, self._ndim))
+        densityMatrixTb = self.makeDensityMatrixTb(mf_model, nK=nK)
         return addTb(
             meanField(densityMatrixTb, self.h_int, n=self._ndim),
             {self._localKey: -self.EF * np.eye(self._size)},