diff --git a/.docs/README.md b/.docs/README.md
index 19350f3..82c97b0 100755
--- a/.docs/README.md
+++ b/.docs/README.md
@@ -51,6 +51,27 @@ nettrine.migrations:
   directories: array<string, string>
   versionsOrganization: <null|year|year_and_month>
   customTemplate: <null|path>
+  allOrNothing: <bool>
+
+  migrationFactory: <service>
+  logger: <service>
+  connection: <string>
+  manager: <string>
+```
+
+**Multiple databases**
+
+```php
+$this->configurator->addDynamicParameters([
+	'env' => getenv(),
+]);
+```
+
+```neon
+nettrine.migrations:
+  directories:
+    App\Migrations: %appDir%/migrations
+  connection: %env.DATABASE_CONNECTION%
 ```
 
 ## Usage
@@ -136,4 +157,3 @@ Since this moment when you type `bin/console`, there'll be registered commands f
 
 > [!TIP]
 > Take a look at more examples in [contributte/doctrine](https://github.com/contributte/doctrine/tree/master/.docs).
-
diff --git a/ruleset.xml b/ruleset.xml
index 697077a..62feb13 100644
--- a/ruleset.xml
+++ b/ruleset.xml
@@ -1,7 +1,9 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <ruleset name="Contributte" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="vendor/squizlabs/php_codesniffer/phpcs.xsd">
 	<!-- Rulesets -->
-	<rule ref="./vendor/contributte/qa/ruleset-8.2.xml"/>
+	<rule ref="./vendor/contributte/qa/ruleset-8.2.xml">
+		<exclude name="SlevomatCodingStandard.Classes.SuperfluousInterfaceNaming.SuperfluousSuffix"/>
+	</rule>
 
 	<!-- Rules -->
 	<rule ref="SlevomatCodingStandard.Files.TypeNameMatchesFileName">
diff --git a/src/DI/Helpers/SmartStatement.php b/src/DI/Helpers/SmartStatement.php
new file mode 100644
index 0000000..1567213
--- /dev/null
+++ b/src/DI/Helpers/SmartStatement.php
@@ -0,0 +1,22 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations\DI\Helpers;
+
+use Nette\DI\Definitions\Statement;
+use Nettrine\Migrations\Exceptions\LogicalException;
+
+final class SmartStatement
+{
+
+	public static function from(mixed $service): Statement
+	{
+		if (is_string($service)) {
+			return new Statement($service);
+		} elseif ($service instanceof Statement) {
+			return $service;
+		} else {
+			throw new LogicalException('Unsupported type of service');
+		}
+	}
+
+}
diff --git a/src/DI/MigrationsExtension.php b/src/DI/MigrationsExtension.php
index a864004..6dd06ce 100644
--- a/src/DI/MigrationsExtension.php
+++ b/src/DI/MigrationsExtension.php
@@ -3,7 +3,7 @@
 namespace Nettrine\Migrations\DI;
 
 use Doctrine\Migrations\Configuration\Configuration;
-use Doctrine\Migrations\DependencyFactory as NativeDependencyFactory;
+use Doctrine\Migrations\DependencyFactory;
 use Doctrine\Migrations\Metadata\Storage\TableMetadataStorageConfiguration;
 use Doctrine\Migrations\Tools\Console\Command\CurrentCommand;
 use Doctrine\Migrations\Tools\Console\Command\DiffCommand;
@@ -18,11 +18,13 @@
 use Doctrine\Migrations\Tools\Console\Command\SyncMetadataCommand;
 use Doctrine\Migrations\Tools\Console\Command\UpToDateCommand;
 use Doctrine\Migrations\Tools\Console\Command\VersionCommand;
+use Doctrine\Migrations\Version\MigrationFactory;
 use Nette\DI\CompilerExtension;
+use Nette\DI\Definitions\Statement;
 use Nette\Schema\Expect;
 use Nette\Schema\Schema;
-use Nettrine\Migrations\DependencyFactory;
-use Nettrine\Migrations\Version\DbalMigrationFactory;
+use Nettrine\Migrations\DependencyFactoryCreator;
+use Nettrine\Migrations\DI\Helpers\SmartStatement;
 use stdClass;
 
 /**
@@ -33,6 +35,11 @@ final class MigrationsExtension extends CompilerExtension
 
 	public function getConfigSchema(): Schema
 	{
+		$expectService = Expect::anyOf(
+			Expect::string()->required()->assert(fn ($input) => str_starts_with($input, '@') || class_exists($input) || interface_exists($input)),
+			Expect::type(Statement::class)->required(),
+		);
+
 		return Expect::structure([
 			'table' => Expect::string('doctrine_migrations'),
 			'column' => Expect::string('version'),
@@ -44,6 +51,10 @@ public function getConfigSchema(): Schema
 			),
 			'customTemplate' => Expect::string(),
 			'allOrNothing' => Expect::bool(false),
+			'logger' => (clone $expectService),
+			'migrationFactory' => (clone $expectService),
+			'connection' => Expect::string(),
+			'manager' => Expect::string(),
 		]);
 	}
 
@@ -66,6 +77,14 @@ public function loadConfiguration(): void
 			->addSetup('setMetadataStorageConfiguration', [$storage])
 			->addSetup('setAllOrNothing', [$config->allOrNothing]);
 
+		if ($config->connection !== null) {
+			$configuration->addSetup('setConnectionName', [$config->connection]);
+		}
+
+		if ($config->manager !== null) {
+			$configuration->addSetup('setEntityManagerName', [$config->manager]);
+		}
+
 		foreach ($config->directories as $namespace => $directory) {
 			$configuration->addSetup('addMigrationsDirectory', [$namespace, $directory]);
 		}
@@ -76,15 +95,13 @@ public function loadConfiguration(): void
 			$configuration->addSetup('setMigrationsAreOrganizedByYearAndMonth');
 		}
 
-		$builder->addDefinition($this->prefix('migrationFactory'))
-			->setFactory(DbalMigrationFactory::class);
-
-		$builder->addDefinition($this->prefix('dependencyFactory'))
-			->setType(NativeDependencyFactory::class)
-			->setFactory($this->prefix('@internal.dependencyFactory::create'));
+		$dependencyFactory = $builder->addDefinition($this->prefix('dependencyFactory'))
+			->setType(DependencyFactory::class)
+			->setFactory(DependencyFactoryCreator::class . '::create');
 
-		$builder->addDefinition($this->prefix('internal.dependencyFactory'))
-			->setFactory(DependencyFactory::class);
+		if ($config->migrationFactory !== null) {
+			$dependencyFactory->addSetup('setService', [MigrationFactory::class, SmartStatement::from($config->migrationFactory)]);
+		}
 
 		// Register commands
 
diff --git a/src/DependencyFactory.php b/src/DependencyFactory.php
deleted file mode 100644
index 8d3ff08..0000000
--- a/src/DependencyFactory.php
+++ /dev/null
@@ -1,51 +0,0 @@
-<?php declare(strict_types = 1);
-
-namespace Nettrine\Migrations;
-
-use Doctrine\DBAL\Connection;
-use Doctrine\Migrations\Configuration\Configuration;
-use Doctrine\Migrations\Configuration\Connection\ExistingConnection;
-use Doctrine\Migrations\Configuration\EntityManager\ExistingEntityManager;
-use Doctrine\Migrations\Configuration\Migration\ExistingConfiguration;
-use Doctrine\Migrations\DependencyFactory as NativeDependencyFactory;
-use Doctrine\Migrations\Version\MigrationFactory;
-use Doctrine\ORM\EntityManagerInterface;
-use Nette\DI\ServiceCreationException;
-use Nettrine\Migrations\Version\DbalMigrationFactory;
-use Psr\Log\LoggerInterface;
-
-class DependencyFactory
-{
-
-	public function __construct(
-		private Configuration $configuration,
-		private DbalMigrationFactory $migrationFactory,
-		private ?Connection $connection = null,
-		private ?EntityManagerInterface $entityManager = null,
-		private ?LoggerInterface $logger = null
-	)
-	{
-	}
-
-	public function create(): NativeDependencyFactory
-	{
-		if ($this->entityManager !== null) {
-			$dependencyFactory = NativeDependencyFactory::fromEntityManager(new ExistingConfiguration($this->configuration), new ExistingEntityManager($this->entityManager), $this->logger);
-		} elseif ($this->connection !== null) {
-			$dependencyFactory = NativeDependencyFactory::fromConnection(new ExistingConfiguration($this->configuration), new ExistingConnection($this->connection), $this->logger);
-		} else {
-			throw new ServiceCreationException(
-				sprintf(
-					'Either service of type %s or %s needs to be registered for Doctrine migrations to work properly',
-					Connection::class,
-					EntityManagerInterface::class
-				)
-			);
-		}
-
-		$dependencyFactory->setService(MigrationFactory::class, $this->migrationFactory);
-
-		return $dependencyFactory;
-	}
-
-}
diff --git a/src/DependencyFactoryCreator.php b/src/DependencyFactoryCreator.php
new file mode 100644
index 0000000..402d1d0
--- /dev/null
+++ b/src/DependencyFactoryCreator.php
@@ -0,0 +1,76 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations;
+
+use Doctrine\Migrations\AbstractMigration;
+use Doctrine\Migrations\Configuration\Configuration;
+use Doctrine\Migrations\Configuration\Connection\ConnectionRegistryConnection;
+use Doctrine\Migrations\Configuration\EntityManager\ManagerRegistryEntityManager;
+use Doctrine\Migrations\Configuration\Migration\ExistingConfiguration;
+use Doctrine\Migrations\DependencyFactory;
+use Doctrine\Migrations\Version\MigrationFactory;
+use Doctrine\Persistence\ConnectionRegistry;
+use Doctrine\Persistence\ManagerRegistry;
+use Nette\DI\Container;
+use Nettrine\Migrations\Exceptions\LogicalException;
+use Nettrine\Migrations\Migration\MigrationFactoryDecorator;
+use Psr\Log\LoggerInterface;
+use Psr\Log\NullLogger;
+
+final class DependencyFactoryCreator
+{
+
+	public static function create(
+		Container $container,
+		Configuration $configuration,
+		?ConnectionRegistry $connectionRegistry = null,
+		?ManagerRegistry $managerRegistry = null,
+		?LoggerInterface $logger = null
+	): DependencyFactory
+	{
+		$logger ??= new NullLogger();
+
+		if ($managerRegistry !== null) {
+			$dependencyFactory = DependencyFactory::fromEntityManager(
+				new ExistingConfiguration($configuration),
+				ManagerRegistryEntityManager::withSimpleDefault($managerRegistry),
+				$logger
+			);
+		} elseif ($connectionRegistry !== null) {
+			$dependencyFactory = DependencyFactory::fromConnection(
+				new ExistingConfiguration($configuration),
+				ConnectionRegistryConnection::withSimpleDefault($connectionRegistry),
+				$logger
+			);
+		} else {
+			throw new LogicalException('You must provide either ManagerRegistry or ConnectionRegistry.');
+		}
+
+		$migrationFactory = new class ($dependencyFactory) implements MigrationFactory {
+
+			public function __construct(
+				private DependencyFactory $dependencyFactory
+			)
+			{
+			}
+
+			public function createVersion(string $migrationClassName): AbstractMigration
+			{
+				$migration = new $migrationClassName(
+					$this->dependencyFactory->getConnection(),
+					$this->dependencyFactory->getLogger()
+				);
+
+				assert($migration instanceof AbstractMigration);
+
+				return $migration;
+			}
+
+		};
+
+		$dependencyFactory->setService(MigrationFactory::class, new MigrationFactoryDecorator($container, $migrationFactory));
+
+		return $dependencyFactory;
+	}
+
+}
diff --git a/src/Exceptions/LogicalException.php b/src/Exceptions/LogicalException.php
new file mode 100644
index 0000000..8d046cb
--- /dev/null
+++ b/src/Exceptions/LogicalException.php
@@ -0,0 +1,8 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations\Exceptions;
+
+final class LogicalException extends \LogicException
+{
+
+}
diff --git a/src/Exceptions/RuntimeException.php b/src/Exceptions/RuntimeException.php
new file mode 100644
index 0000000..cbd485e
--- /dev/null
+++ b/src/Exceptions/RuntimeException.php
@@ -0,0 +1,8 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations\Exceptions;
+
+final class RuntimeException extends \RuntimeException
+{
+
+}
diff --git a/src/Migration/ContainerAwareInterface.php b/src/Migration/ContainerAwareInterface.php
new file mode 100644
index 0000000..971a74c
--- /dev/null
+++ b/src/Migration/ContainerAwareInterface.php
@@ -0,0 +1,12 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations\Migration;
+
+use Nette\DI\Container;
+
+interface ContainerAwareInterface
+{
+
+	public function setContainer(Container $container): void;
+
+}
diff --git a/src/Migration/MigrationFactoryDecorator.php b/src/Migration/MigrationFactoryDecorator.php
new file mode 100644
index 0000000..bbddb5e
--- /dev/null
+++ b/src/Migration/MigrationFactoryDecorator.php
@@ -0,0 +1,34 @@
+<?php declare(strict_types = 1);
+
+namespace Nettrine\Migrations\Migration;
+
+use Doctrine\Migrations\AbstractMigration;
+use Doctrine\Migrations\Version\MigrationFactory;
+use Nette\DI\Container;
+
+class MigrationFactoryDecorator implements MigrationFactory
+{
+
+	public function __construct(
+		private Container $container,
+		private MigrationFactory $migrationFactory,
+	)
+	{
+	}
+
+	public function createVersion(string $migrationClassName): AbstractMigration
+	{
+		$instance = $this->migrationFactory->createVersion($migrationClassName);
+
+		// Call setContainer
+		if ($instance instanceof ContainerAwareInterface) {
+			$instance->setContainer($this->container);
+		}
+
+		// Allow to use inject<>
+		$this->container->callInjects($instance);
+
+		return $instance;
+	}
+
+}
diff --git a/src/Version/DbalMigrationFactory.php b/src/Version/DbalMigrationFactory.php
deleted file mode 100644
index 4be8de5..0000000
--- a/src/Version/DbalMigrationFactory.php
+++ /dev/null
@@ -1,42 +0,0 @@
-<?php declare(strict_types = 1);
-
-namespace Nettrine\Migrations\Version;
-
-use Doctrine\DBAL\Connection;
-use Doctrine\Migrations\AbstractMigration;
-use Doctrine\Migrations\Version\MigrationFactory;
-use Nette\DI\Container;
-use Psr\Log\LoggerInterface;
-use Psr\Log\NullLogger;
-
-class DbalMigrationFactory implements MigrationFactory
-{
-
-	private Container $container;
-
-	private Connection $connection;
-
-	private LoggerInterface $logger;
-
-	public function __construct(Container $container, Connection $connection, ?LoggerInterface $logger = null)
-	{
-		$this->container = $container;
-		$this->connection = $connection;
-		$this->logger = $logger ?? new NullLogger();
-	}
-
-	public function createVersion(string $migrationClassName): AbstractMigration
-	{
-		$migration = new $migrationClassName(
-			$this->connection,
-			$this->logger
-		);
-
-		assert($migration instanceof AbstractMigration);
-
-		$this->container->callInjects($migration);
-
-		return $migration;
-	}
-
-}
diff --git a/tests/Cases/DI/MigrationsExtension.phpt b/tests/Cases/DI/MigrationsExtension.phpt
index 3160dba..3b0ed0a 100644
--- a/tests/Cases/DI/MigrationsExtension.phpt
+++ b/tests/Cases/DI/MigrationsExtension.phpt
@@ -10,26 +10,27 @@ use Doctrine\Migrations\DependencyFactory;
 use Doctrine\Migrations\Tools\Console\Command\DoctrineCommand;
 use Nette\DI\Compiler;
 use Nettrine\Migrations\DI\MigrationsExtension;
-use Nettrine\Migrations\Version\DbalMigrationFactory;
+use Nettrine\Migrations\Exceptions\LogicalException;
+use Nettrine\Migrations\Migration\MigrationFactoryDecorator;
 use Symfony\Component\Console\Application;
 use Tester\Assert;
 
 require_once __DIR__ . '/../../bootstrap.php';
 
+// OK
 Toolkit::test(function (): void {
 	$container = ContainerBuilder::of()
 		->withCompiler(function (Compiler $compiler): void {
 			$compiler->addExtension('migrations', new MigrationsExtension());
 			$compiler->addConfig(Neonkit::load('
-				parameters:
-					appDir: /root
 				migrations:
 					directories:
-						App\Domain: %appDir%/migrations
+						App\Domain: /root/migrations
 				services:
 					- Symfony\Component\Console\Application
 					- Doctrine\DBAL\Driver\Mysqli\Driver
 					- Doctrine\DBAL\Connection([])
+					- Tests\Mocks\DummyConnectionRegistry
 			'));
 		})
 		->build();
@@ -45,5 +46,23 @@ Toolkit::test(function (): void {
 	/** @var DependencyFactory $dependencyFactory */
 	$dependencyFactory = $container->getByType(DependencyFactory::class);
 	Assert::type(DependencyFactory::class, $dependencyFactory);
-	Assert::type(DbalMigrationFactory::class, $dependencyFactory->getMigrationFactory());
+	Assert::type(MigrationFactoryDecorator::class, $dependencyFactory->getMigrationFactory());
+});
+
+// No ConnectionRegistry or ManagerRegistry
+Toolkit::test(function (): void {
+	Assert::exception(function (): void {
+		$container = ContainerBuilder::of()
+			->withCompiler(function (Compiler $compiler): void {
+				$compiler->addExtension('migrations', new MigrationsExtension());
+				$compiler->addConfig(Neonkit::load('
+				migrations:
+					directories:
+						App\Domain: /root/migrations
+			'));
+			})
+			->build();
+
+		$container->getByType(DependencyFactory::class);
+	}, LogicalException::class, 'You must provide either ManagerRegistry or ConnectionRegistry.');
 });
diff --git a/tests/Mocks/DummyConnectionRegistry.php b/tests/Mocks/DummyConnectionRegistry.php
new file mode 100644
index 0000000..f18e5ee
--- /dev/null
+++ b/tests/Mocks/DummyConnectionRegistry.php
@@ -0,0 +1,43 @@
+<?php declare(strict_types = 1);
+
+namespace Tests\Mocks;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\Persistence\ConnectionRegistry;
+
+final class DummyConnectionRegistry implements ConnectionRegistry
+{
+
+	public function __construct(
+		private Connection $connection
+	)
+	{
+	}
+
+	public function getDefaultConnectionName(): string
+	{
+		return 'default';
+	}
+
+	public function getConnection(?string $name = null): Connection
+	{
+		return $this->connection;
+	}
+
+	/**
+	 * @return array<string, Connection>
+	 */
+	public function getConnections(): array
+	{
+		return ['default' => $this->connection];
+	}
+
+	/**
+	 * @return string[]
+	 */
+	public function getConnectionNames(): array
+	{
+		return ['default'];
+	}
+
+}