From 039d7f01a1264897255b02b59941857103891088 Mon Sep 17 00:00:00 2001
From: Samuel Williams <samuel.williams@oriontransfer.co.nz>
Date: Thu, 8 Aug 2024 23:42:58 +1200
Subject: [PATCH] Fix tests.

---
 test/console.rb        | 2 +-
 test/console/logger.rb | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/console.rb b/test/console.rb
index 855e88a..7aab3af 100644
--- a/test/console.rb
+++ b/test/console.rb
@@ -114,7 +114,7 @@ def before
 			super
 		end
 		
-		def after
+		def after(error = nil)
 			subject.logger = @original_logger
 			
 			super
diff --git a/test/console/logger.rb b/test/console/logger.rb
index 96fa628..bdd4eca 100644
--- a/test/console/logger.rb
+++ b/test/console/logger.rb
@@ -124,7 +124,7 @@ def before
 			super
 		end
 		
-		def after
+		def after(error = nil)
 			$DEBUG = @debug
 			$VERBOSE = @verbose
 			super