1 files changed, 3 insertions, 2 deletions
diff --git a/register.php b/register.php
index 44ab879..e286f01 100755
--- a/register.php
+++ b/register.php
@@ -32,12 +32,13 @@ if ($_SERVER['REQUEST_METHOD'] == 'POST') {
 	$invite_code = $form->validate_invite_code($_POST['invite_code']);
 
     $form->on_success(function () use ($username, $password, $invite_code) {
-        User::register($username, $password);
-
         $inviter = new User();
         $inviter->get_by_invite_code($invite_code);
         $inviter->update_invite_code();
 
+        $generation = $inviter->generation + 1;
+        User::register($username, $password, $generation);
+
         echo '<p class="success">Account successfully registered! You can now <a href="signin.php">sign in</a></p>';
     });
 }