From 37193b1f5b03078855156c60d9923af869827cf4 Mon Sep 17 00:00:00 2001
From: Mitchell Lutzke <mitchell649@hotmail.com>
Date: Sat, 26 Oct 2024 21:36:43 -0700
Subject: [PATCH] slight cleanup

---
 .../unittests/jackson/TestChatRequestSerialization.java      | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java b/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java
index 80cb063..2ce210c 100644
--- a/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java
+++ b/src/test/java/io/github/ollama4j/unittests/jackson/TestChatRequestSerialization.java
@@ -1,14 +1,13 @@
 package io.github.ollama4j.unittests.jackson;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrowsExactly;
 
 import java.io.File;
-import java.security.InvalidParameterException;
 import java.util.List;
 
 import io.github.ollama4j.models.chat.OllamaChatRequest;
 import org.json.JSONObject;
-import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
@@ -87,7 +86,7 @@ public class TestChatRequestSerialization extends AbstractSerializationTest<Olla
     @Test
     public void testRequestWithInvalidCustomOption() {
         OptionsBuilder b = new OptionsBuilder();
-        Assertions.assertThrowsExactly(IllegalArgumentException.class, () -> {
+        assertThrowsExactly(IllegalArgumentException.class, () -> {
                 OllamaChatRequest req = builder.withMessage(OllamaChatMessageRole.USER, "Some prompt")
                 .withOptions(b.setCustomOption("cust_obj", new Object()).build())
                 .build();