diff --git a/src/Http/Controllers/TeamInvitationController.php b/src/Http/Controllers/TeamInvitationController.php
index 2331be071..2bfcb09d1 100644
--- a/src/Http/Controllers/TeamInvitationController.php
+++ b/src/Http/Controllers/TeamInvitationController.php
@@ -7,7 +7,7 @@
 use Illuminate\Routing\Controller;
 use Illuminate\Support\Facades\Gate;
 use Laravel\Jetstream\Contracts\AddsTeamMembers;
-use Laravel\Jetstream\Jetstream;
+use Laravel\Jetstream\TeamInvitation;
 
 class TeamInvitationController extends Controller
 {
@@ -15,15 +15,11 @@ class TeamInvitationController extends Controller
      * Accept a team invitation.
      *
      * @param  \Illuminate\Http\Request  $request
-     * @param  int  $invitationId
+     * @param  \Laravel\Jetstream\TeamInvitation  $invitation
      * @return \Illuminate\Http\RedirectResponse
      */
-    public function accept(Request $request, $invitationId)
+    public function accept(Request $request, TeamInvitation $invitation)
     {
-        $model = Jetstream::teamInvitationModel();
-
-        $invitation = $model::whereKey($invitationId)->firstOrFail();
-
         app(AddsTeamMembers::class)->add(
             $invitation->team->owner,
             $invitation->team,
@@ -42,15 +38,11 @@ public function accept(Request $request, $invitationId)
      * Cancel the given team invitation.
      *
      * @param  \Illuminate\Http\Request  $request
-     * @param  int  $invitationId
+     * @param  \Laravel\Jetstream\TeamInvitation  $invitation
      * @return \Illuminate\Http\RedirectResponse
      */
-    public function destroy(Request $request, $invitationId)
+    public function destroy(Request $request, TeamInvitation $invitation)
     {
-        $model = Jetstream::teamInvitationModel();
-
-        $invitation = $model::whereKey($invitationId)->firstOrFail();
-
         if (! Gate::forUser($request->user())->check('removeTeamMember', $invitation->team)) {
             throw new AuthorizationException;
         }