diff --git a/did/views.py b/did/views.py
index c53d47f..9294e0b 100644
--- a/did/views.py
+++ b/did/views.py
@@ -106,7 +106,7 @@ class PublicDeviceWebView(TemplateView):
             'device': {},
         }
         dev = Build(self.object.last_evidence.doc, None, check=True)
-        doc = dev.get_doc()
+        doc = dev.build.get_doc()
         data['document'] = json.dumps(doc)
         data['device'] = doc.device
         data['components'] = doc.components
diff --git a/dpp/views.py b/dpp/views.py
index 13753c2..27ca1e6 100644
--- a/dpp/views.py
+++ b/dpp/views.py
@@ -12,7 +12,7 @@ from dpp.models import Proof
 
 
 class ProofView(View):
-    
+
     def get(self, request, *args, **kwargs):
         timestamp = kwargs.get("proof_id")
         proof = Proof.objects.filter(timestamp=timestamp).first()
@@ -22,9 +22,9 @@ class ProofView(View):
         ev = Evidence(proof.uuid)
         if not ev.doc:
             return JsonResponse({}, status=404)
-        
+
         dev = Build(ev.doc, None, check=True)
-        doc = dev.get_phid()
+        doc = dev.build.get_doc()
 
         data = {
             "algorithm": ALGORITHM,