diff --git a/gpt4all-chat/qml/ChatView.qml b/gpt4all-chat/qml/ChatView.qml
index 38bac37c..f028bf9d 100644
--- a/gpt4all-chat/qml/ChatView.qml
+++ b/gpt4all-chat/qml/ChatView.qml
@@ -1170,7 +1170,7 @@ Rectangle {
                                     states: [
                                         State {
                                             name: "expanded"
-                                            PropertyChanges { target: sourcesLayout; Layout.preferredHeight: childrenRect.height }
+                                            PropertyChanges { target: sourcesLayout; Layout.preferredHeight: flow.height }
                                         },
                                         State {
                                             name: "collapsed"
@@ -1192,6 +1192,7 @@ Rectangle {
                                     ]
 
                                     Flow {
+                                        id: flow
                                         Layout.fillWidth: true
                                         spacing: 10
                                         visible: consolidatedSources.length !== 0