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