1 files changed, 5 insertions, 1 deletions
diff --git a/test.sh b/test.sh
index 1f8f11e..a7ba75d 100755
--- a/test.sh
+++ b/test.sh
@@ -6,7 +6,11 @@
 ##################################################
 
 function java() {
-    expected=$(bash runner.sh java "$1.java")
+    base="$1"
+    if [[ $(basename $base) == *.* ]]; then
+        base="${base%.*}"
+    fi
+    expected=$(bash runner.sh java "${base}.java")
 }
 
 function match() {