diff --git a/com_leaflet/site/src/Model/LeafletModel.php b/com_leaflet/site/src/Model/LeafletModel.php
index 30dcd64..27a54b1 100644
--- a/com_leaflet/site/src/Model/LeafletModel.php
+++ b/com_leaflet/site/src/Model/LeafletModel.php
@@ -50,7 +50,7 @@ class LeafletModel extends ItemModel {
 	{
 		try 
 		{
-			$db    = Factory::getDbo();
+			$db = Factory::getContainer()->get('DatabaseDriver');
 			$query = $db
 				->getQuery(true)
 				->select('id')
@@ -79,7 +79,7 @@ class LeafletModel extends ItemModel {
 	{
 		try 
 		{
-			$db    = Factory::getDbo();
+			$db = Factory::getContainer()->get('DatabaseDriver');
 			$query = $db->getQuery(true);
 
 			$query->select($db->quoteName(array('id', 'title', 'name', 'fieldparams')));
@@ -102,7 +102,7 @@ class LeafletModel extends ItemModel {
 	{
 		try 
 		{
-			$db    = Factory::getDbo();
+			$db = Factory::getContainer()->get('DatabaseDriver');
 			$query = $db->getQuery(true);
 
 			$query->select($db->quoteName(array('id', 'title', 'name', 'fieldparams')));
@@ -127,7 +127,7 @@ class LeafletModel extends ItemModel {
 	{
 		try 
 		{
-			$db    = Factory::getDbo();
+			$db = Factory::getContainer()->get('DatabaseDriver');
 			$query = $db->getQuery(true);
 
 			$query->select($db->quoteName(array('id', 'title', 'name', 'fieldparams')));
@@ -152,7 +152,7 @@ class LeafletModel extends ItemModel {
 	{
 		try 
 		{
-			$db    = Factory::getDbo();
+			$db = Factory::getContainer()->get('DatabaseDriver');
 
 			$query = $db->getQuery(true);
 			$groupId = 4;
@@ -171,6 +171,7 @@ class LeafletModel extends ItemModel {
 			'c.title',
 			'c.alias',
 			'c.state',
+			'c.catid',
 			'c.introtext');
 			foreach($fields as $field) {
 				array_push($fieldsSelect, 'GROUP_CONCAT(DISTINCT IF(field_id = ' . $field->id . ', value, NULL) SEPARATOR ",") AS ' . $field->name);
@@ -205,4 +206,4 @@ class LeafletModel extends ItemModel {
 
 		return $results; 
 	}
-}
\ No newline at end of file
+}