Corrects setup file.

Underpin-WP · Jul 12, 2021 · d1fde3d · d1fde3d
1 parent 40cf6a9
commit d1fde3d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/underpin-berlin-db.php b/underpin-berlin-db.php
@@ -3,7 +3,7 @@
 	exit;
 }
 
-add_action( 'underpin/before_setup', function ( $instance ) {
+add_action( 'underpin/before_setup', function ( $file, $class ) {
 	if ( ! defined( 'UNDERPIN_BERLIN_DB_PATH' ) ) {
 		define( 'UNDERPIN_BERLIN_DB_PATH', trailingslashit( __DIR__ ) );
 	}
@@ -13,7 +13,7 @@
 	require_once( UNDERPIN_BERLIN_DB_PATH . 'lib/factories/Database_Model_Instance.php' );
 	require_once( UNDERPIN_BERLIN_DB_PATH . 'lib/factories/Database_Model_With_Meta_Instance.php' );
 
-	$instance->loaders()->add( 'berlin_db', [
+	Underpin\underpin()->get( $file, $class )->loaders()->add( 'berlin_db', [
 		'registry' => 'Underpin_BerlinDB\Loaders\Database'
 	] );
-}, 20 );
+}, 20, 2 );