solved s06 ex05

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;
     }