closed #78

Shared/views/ContentView.swift

@@ -71,6 +71,7 @@ struct ContentView: View {
                         viewModel.update(entry: selectedEntry, toMood: mood)
                     }
                     showUpdateEntryAlert = false
+                    selectedEntry = nil
                 })
             }
             
@@ -279,7 +280,7 @@ struct ContentView: View {
             if viewModel.shouldShowVotingHeader() {
                 AddMoodHeaderView(addItemHeaderClosure: { (mood, date) in
                     withAnimation {
-                        viewModel.add(mood: mood, forDate: date)
+                        viewModel.add(mood: mood, forDate: date, entryType: .header)
                     }
                 })
                     .frame(height: headerHeight)
@@ -319,7 +320,7 @@ struct ContentView: View {
                     .padding()
                 AddMoodHeaderView(addItemHeaderClosure: { (mood, date) in
                     withAnimation {
-                        viewModel.add(mood: mood, forDate: date)
+                        viewModel.add(mood: mood, forDate: date, entryType: .header)
                     }
                 }, overrideDay: viewModel.shouldShowVotingHeader() ? .Today : .Previous)
             }