fix: domain return MockUnitString

to use MockUnitString chain
nomemory · Dec 29, 2021 · 3a86b1e · 3a86b1e
1 parent 6f92799
commit 3a86b1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/net/andreinc/mockneat/unit/user/Emails.java b/src/main/java/net/andreinc/mockneat/unit/user/Emails.java
@@ -36,7 +36,7 @@ public Supplier<String> supplier() {
      * @param domains A var-arg String array containing the list of the domains to be used.
      * @return A new {@code MockUnitString}.
      */
-    public MockUnit<String> domains(String... domains) {
+    public MockUnitString domains(String... domains) {
         Supplier<String> supp = () -> {
             String user = mockNeat.users().val();
             String domain = mockNeat.from(domains).val();
@@ -53,7 +53,7 @@ public MockUnit<String> domains(String... domains) {
      * @param domain The domain to be used.
      * @return A new {@code MockUnitString}.
      */
-    public MockUnit<String> domain(String domain) {
+    public MockUnitString domain(String domain) {
         return domains(domain);
     }
 }