diff --git a/src/main/java/org/apache/flink/benchmark/StateBackendBenchmarkBase.java b/src/main/java/org/apache/flink/benchmark/StateBackendBenchmarkBase.java
index cbbbd05..7bf24c2 100644
--- a/src/main/java/org/apache/flink/benchmark/StateBackendBenchmarkBase.java
+++ b/src/main/java/org/apache/flink/benchmark/StateBackendBenchmarkBase.java
@@ -87,7 +87,9 @@
 			source = env.addSource(new IntegerLongSource(numberOfElements, recordsPerInvocation));
 		}
 
-		public void tearDown() throws IOException {
+		@Override
+		public void tearDown() throws Exception {
+			super.tearDown();
 			FileUtils.deleteDirectory(checkpointDir);
 		}
 	}