diff --git a/apps/probability/calculation_controller.cpp b/apps/probability/calculation_controller.cpp
index 96e7957fb..caf917d66 100644
--- a/apps/probability/calculation_controller.cpp
+++ b/apps/probability/calculation_controller.cpp
@@ -175,7 +175,6 @@ bool CalculationController::handleEvent(Ion::Events::Event event) {
     if (m_highlightedSubviewIndex == 0) {
       m_contentView.imageTableView()->select(false);
       m_contentView.imageTableView()->setHighlight(false);
-      m_contentView.layoutSubviews();
     } else {
       EditableTextCell * calculCell = m_contentView.calculationCellAtIndex(m_highlightedSubviewIndex-1);
       calculCell->setHighlighted(false);
diff --git a/apps/probability/image_table_view.cpp b/apps/probability/image_table_view.cpp
index 5b7ebd7f5..21f1c780d 100644
--- a/apps/probability/image_table_view.cpp
+++ b/apps/probability/image_table_view.cpp
@@ -80,6 +80,10 @@ void ImageTableView::didBecomeFirstResponder() {
   app()->setFirstResponder(&m_selectableTableView);
 }
 
+void ImageTableView::willExitResponderChain(Responder * nextFirstResponder) {
+  m_calculationController->reload();
+}
+
 bool ImageTableView::handleEvent(Ion::Events::Event event) {
   if (event == Ion::Events::OK) {
     m_calculationController->setCalculationAccordingToIndex(m_selectableTableView.selectedRow());
diff --git a/apps/probability/image_table_view.h b/apps/probability/image_table_view.h
index 56009097b..1e10685dd 100644
--- a/apps/probability/image_table_view.h
+++ b/apps/probability/image_table_view.h
@@ -32,6 +32,7 @@ class ImageTableView : public View, public Responder, public SimpleListViewDataS
   void setCalculation(Calculation * calculation, int index);
   bool handleEvent(Ion::Events::Event event) override;
   void didBecomeFirstResponder() override;
+  void willExitResponderChain(Responder * nextFirstResponder) override;
   void select(bool select);
   void setHighlight(bool highlight);
   int numberOfRows() override;