Index: trunk/psphot/src/psphotSourceFits.c
===================================================================
--- trunk/psphot/src/psphotSourceFits.c	(revision 16820)
+++ trunk/psphot/src/psphotSourceFits.c	(revision 17396)
@@ -220,5 +220,5 @@
     pmSource *tmpSrc = pmSourceAlloc ();
 
-    pmModel *EXT = psphotFitEXT (readout, source, maskVal);
+    pmModel *EXT = psphotFitEXT (readout, source, modelTypeEXT, maskVal);
     okEXT = psphotEvalEXT (tmpSrc, EXT);
     chiEXT = EXT->chisq / EXT->nDOF;
@@ -267,4 +267,6 @@
     // save new model
     source->modelEXT = EXT;
+    source->type = PM_SOURCE_TYPE_EXTENDED;
+    source->mode |= PM_SOURCE_MODE_EXTMODEL;
 
     // build cached model and subtract
@@ -349,10 +351,10 @@
 }
 
-pmModel *psphotFitEXT (pmReadout *readout, pmSource *source, psMaskType maskVal) {
+pmModel *psphotFitEXT (pmReadout *readout, pmSource *source, pmModelType modelType, psMaskType maskVal) {
 
     NfitEXT ++;
 
     // use the source moments, etc to guess basic model parameters
-    pmModel *EXT = pmSourceModelGuess (source, modelTypeEXT);
+    pmModel *EXT = pmSourceModelGuess (source, modelType);
     PS_ASSERT (EXT, NULL);