Allow &self without a type when the block only has one type
diff --git a/tests/test.rs b/tests/test.rs
index eb14d36..1cd85df 100644
--- a/tests/test.rs
+++ b/tests/test.rs
@@ -102,8 +102,8 @@
     assert_eq!(2020, old_value);
     assert_eq!(2021, unique_ptr.set(2021));
     assert_eq!(2021, unique_ptr.get());
-    assert_eq!(old_value, unique_ptr.set(old_value));
-    assert_eq!(old_value, unique_ptr.get())
+    assert_eq!(old_value, unique_ptr.set2(old_value));
+    assert_eq!(old_value, unique_ptr.get2())
 }
 
 #[no_mangle]