diff --git a/example/lib/example_registration_repository.dart b/example/lib/example_registration_repository.dart
index d70677c..41e8064 100644
--- a/example/lib/example_registration_repository.dart
+++ b/example/lib/example_registration_repository.dart
@@ -1,10 +1,12 @@
+import 'dart:collection';
+
 import 'package:flutter/material.dart';
 import 'package:flutter_registration/flutter_registration.dart';
 
 class ExampleRegistrationRepository with RegistrationRepository {
   @override
-  Future<void> register(String email, String password) {
-    debugPrint('register: $email $password');
+  Future<void> register(HashMap values) {
+    debugPrint('register: $values');
     return Future.value(null);
   }
 }
diff --git a/lib/src/registration_screen.dart b/lib/src/registration_screen.dart
index 02f1322..94be138 100644
--- a/lib/src/registration_screen.dart
+++ b/lib/src/registration_screen.dart
@@ -1,7 +1,6 @@
 import 'package:flutter/material.dart';
 import 'package:flutter_registration/flutter_registration.dart';
 import 'package:flutter_registration/src/auth_screen.dart';
-import 'package:flutter_registration/src/model/auth_text_field.dart';
 
 class RegistrationScreen extends StatelessWidget {
   const RegistrationScreen({
@@ -30,8 +29,8 @@ class RegistrationScreen extends StatelessWidget {
           ),
         );
 
-    void register(String email, String password) => repository
-            .register(email, password)
+    void register(values) => repository
+            .register(values)
             .then(
               (value) => afterRegistration(),
             )
@@ -79,10 +78,7 @@ class RegistrationScreen extends StatelessWidget {
         ),
         ...additionalSteps
       ],
-      onFinish: (values) => register(
-        values['email']!,
-        values['password']!,
-      ),
+      onFinish: register,
     );
   }
 }
diff --git a/lib/src/service/registration_repository.dart b/lib/src/service/registration_repository.dart
index 05a920e..9354855 100644
--- a/lib/src/service/registration_repository.dart
+++ b/lib/src/service/registration_repository.dart
@@ -1,3 +1,5 @@
+import 'dart:collection';
+
 mixin RegistrationRepository {
-  Future<void> register(String email, String password);
+  Future<void> register(HashMap values);
 }