diff --git a/extensions/lock/migrations/2015_02_24_000000_add_locked_to_discussions.php b/extensions/lock/migrations/2015_02_24_000000_add_locked_to_discussions.php
index d4f9ae535..eda1c24cd 100644
--- a/extensions/lock/migrations/2015_02_24_000000_add_locked_to_discussions.php
+++ b/extensions/lock/migrations/2015_02_24_000000_add_locked_to_discussions.php
@@ -1,7 +1,7 @@
 <?php
 
 use Illuminate\Database\Schema\Blueprint;
-use Illuminate\Database\Migrations\Migration;
+use Flarum\Migrations\Migration;
 
 class AddLockedToDiscussions extends Migration
 {
@@ -12,7 +12,7 @@ class AddLockedToDiscussions extends Migration
      */
     public function up()
     {
-        Schema::table('discussions', function (Blueprint $table) {
+        $this->schema->table('discussions', function (Blueprint $table) {
             $table->boolean('is_locked')->default(0);
         });
     }
@@ -24,7 +24,7 @@ class AddLockedToDiscussions extends Migration
      */
     public function down()
     {
-        Schema::table('discussions', function (Blueprint $table) {
+        $this->schema->table('discussions', function (Blueprint $table) {
             $table->dropColumn('is_locked');
         });
     }