Bug fix for Query

Sammaye · Jan 9, 2017 · cbed71c · cbed71c
1 parent e539d6f
commit cbed71c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Query.php b/Query.php
@@ -336,7 +336,7 @@ public function getDbConnection()
 	public function setDb($db)
 	{
 		if($db instanceof Database){
-			$this->databaseName = $db->database->getDatabaseName();
+			$this->databaseName = $db->getDatabaseName();
 		}elseif($db instanceof MongoDatabase){
 			$this->databaseName = $db->getDatabaseName();
 		}else{
@@ -349,7 +349,7 @@ public function getDb()
 		$name = $this->databaseName;
 		if($this->model){
 			// Get DB name from model
-			$name = $this->model->getCollection()->database->getDatabaseName();
+			$name = $this->model->getCollection()->getDatabaseName();
 		}
 
 		// Will select default database if name is null