diff --git a/extensions/mentions/migrations/2015_05_11_000000_create_mentions_posts_table.php b/extensions/mentions/migrations/2015_05_11_000000_create_mentions_posts_table.php
index b41a5564e..ca619d0b9 100644
--- a/extensions/mentions/migrations/2015_05_11_000000_create_mentions_posts_table.php
+++ b/extensions/mentions/migrations/2015_05_11_000000_create_mentions_posts_table.php
@@ -1,7 +1,7 @@
 <?php
 
 use Illuminate\Database\Schema\Blueprint;
-use Illuminate\Database\Migrations\Migration;
+use Flarum\Migrations\Migration;
 
 class CreateMentionsPostsTable extends Migration
 {
@@ -12,7 +12,7 @@ class CreateMentionsPostsTable extends Migration
      */
     public function up()
     {
-        Schema::create('mentions_posts', function (Blueprint $table) {
+        $this->schema->create('mentions_posts', function (Blueprint $table) {
             $table->integer('post_id')->unsigned();
             $table->integer('mentions_id')->unsigned();
             $table->primary(['post_id', 'mentions_id']);
@@ -26,6 +26,6 @@ class CreateMentionsPostsTable extends Migration
      */
     public function down()
     {
-        Schema::drop('mentions_posts');
+        $this->schema->drop('mentions_posts');
     }
 }
diff --git a/extensions/mentions/migrations/2015_05_11_000000_create_mentions_users_table.php b/extensions/mentions/migrations/2015_05_11_000000_create_mentions_users_table.php
index 29307fa53..d54223176 100644
--- a/extensions/mentions/migrations/2015_05_11_000000_create_mentions_users_table.php
+++ b/extensions/mentions/migrations/2015_05_11_000000_create_mentions_users_table.php
@@ -1,7 +1,7 @@
 <?php
 
 use Illuminate\Database\Schema\Blueprint;
-use Illuminate\Database\Migrations\Migration;
+use Flarum\Migrations\Migration;
 
 class CreateMentionsUsersTable extends Migration
 {
@@ -12,7 +12,7 @@ class CreateMentionsUsersTable extends Migration
      */
     public function up()
     {
-        Schema::create('mentions_users', function (Blueprint $table) {
+        $this->schema->create('mentions_users', function (Blueprint $table) {
             $table->integer('post_id')->unsigned();
             $table->integer('mentions_id')->unsigned();
             $table->primary(['post_id', 'mentions_id']);
@@ -26,6 +26,6 @@ class CreateMentionsUsersTable extends Migration
      */
     public function down()
     {
-        Schema::drop('mentions_users');
+        $this->schema->drop('mentions_users');
     }
 }