diff --git a/extensions/likes/migrations/2015_05_11_000000_create_posts_likes_table.php b/extensions/likes/migrations/2015_05_11_000000_create_posts_likes_table.php
index fb3693229..e4ef11649 100644
--- a/extensions/likes/migrations/2015_05_11_000000_create_posts_likes_table.php
+++ b/extensions/likes/migrations/2015_05_11_000000_create_posts_likes_table.php
@@ -1,7 +1,7 @@
 <?php
 
 use Illuminate\Database\Schema\Blueprint;
-use Illuminate\Database\Migrations\Migration;
+use Flarum\Migrations\Migration;
 
 class CreatePostsLikesTable extends Migration
 {
@@ -12,7 +12,7 @@ class CreatePostsLikesTable extends Migration
      */
     public function up()
     {
-        app('db')->getSchemaBuilder()->create('posts_likes', function (Blueprint $table) {
+        $this->schema->create('posts_likes', function (Blueprint $table) {
             $table->integer('post_id')->unsigned();
             $table->integer('user_id')->unsigned();
             $table->primary(['post_id', 'user_id']);
@@ -26,6 +26,6 @@ class CreatePostsLikesTable extends Migration
      */
     public function down()
     {
-        app('db')->getSchemaBuilder()->drop('posts_likes');
+        $this->schema->drop('posts_likes');
     }
 }