diff --git a/server/images.go b/server/images.go
index 54bbf158..cd732d7b 100644
--- a/server/images.go
+++ b/server/images.go
@@ -51,7 +51,7 @@ func (m *Model) Prompt(request api.GenerateRequest) (string, error) {
 		Context []int
 	}
 
-	vars.First = len(vars.Context) == 0
+	vars.First = len(request.Context) == 0
 	vars.System = m.System
 	vars.Prompt = request.Prompt
 	vars.Context = request.Context