diff --git a/person.py b/person.py
index d203018..95694aa 100644
--- a/person.py
+++ b/person.py
@@ -87,7 +87,7 @@ def AddRefimgToPerson( filename, person ):
         #False == dont autorotate, its not needed on this image
         refimg.thumbnail, refimg.orig_w, refimg.orig_h = GenThumb( filename, False )
         settings = Settings.query.first()
-        model=AIModel.query.get(settings.default_refimg_model)
+        model=db.session.get(AIModel,settings.default_refimg_model)
         refimg.face, face_locn = GenFace( filename, model=model.name )
         try:
             os.remove(filename)
@@ -194,10 +194,10 @@ def match_with_create_person():
 def person(id):
     form = PersonForm(request.form)
     page_title='Edit Person'
-    
+
     if request.method == 'POST':
         try:
-            person = Person.query.get(id)
+            person = db.session.get(Person, id)
             if 'delete' in request.form:
                 SetFELog( f"Successfully deleted Person: ({person.tag})" )
 
@@ -237,7 +237,7 @@ def person(id):
             SetFELog( f"<b>Failed to modify Person:</b>&nbsp;{e}", "danger" )
             return redirect( url_for( 'persons' ) )
     else:
-        person = Person.query.get(id)
+        person = db.session.get(Person, id)
         if not person:
             SetFELog( f"No such person with id: {id}", "danger" )
             return redirect("/")