From abf20083d1b725bed67931259a780006b1a5e3f9 Mon Sep 17 00:00:00 2001
From: "navid.sassan" <navid.sassan@linuxfabrik.ch>
Date: Mon, 2 Nov 2020 17:08:30 +0100
Subject: [PATCH] solved s06 ex05

---
 06/src/main/java/ch/zhaw/ads/AVLSearchTree.java | 1 +
 1 file changed, 1 insertion(+)

diff --git a/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java b/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java
index 89022b1..405bf2e 100644
--- a/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java
+++ b/06/src/main/java/ch/zhaw/ads/AVLSearchTree.java
@@ -124,6 +124,7 @@ public class AVLSearchTree<T extends Comparable<T>> extends SortedBinaryTree<T>
     public T remove(T x) {
         TreeNode<T> removed = new TreeNode<T>(null);
         root = removeAt(root, x, removed);
+        root = balance(root);
         return removed.element;
     }