diff --git a/code/forms/gridfield/GridFieldConfig_BlogPost.php b/code/forms/gridfield/GridFieldConfig_BlogPost.php
index 4569ff2..776449c 100644
--- a/code/forms/gridfield/GridFieldConfig_BlogPost.php
+++ b/code/forms/gridfield/GridFieldConfig_BlogPost.php
@@ -12,8 +12,8 @@ class GridFieldConfig_BlogPost extends GridFieldConfig_Lumberjack {
 	
 	public function __construct($itemsPerPage = null) {
 		parent::__construct($itemsPerPage);
-		$this->removeComponentsByType('SiteTreeEditButton');
-		$this->addComponent(new GridFieldSiteTreeEditButton());
+		$this->removeComponentsByType('GridFieldSiteTreeState');
+		$this->addComponent(new GridFieldBlogPostState());
 	}
 
 }
\ No newline at end of file
diff --git a/code/model/Blog.php b/code/model/Blog.php
index 1c6aef3..a48b3ab 100644
--- a/code/model/Blog.php
+++ b/code/model/Blog.php
@@ -69,11 +69,6 @@ class Blog extends Page {
 		});
 		
 		$fields = parent::getCMSFields();
-
-		// Ensure we're using the BlogPost GridField config and not Lumberjack's
-		$gridField = $fields->dataFieldByName('ChildPages');
-		$gridField->setConfig(GridFieldConfig_BlogPost::create());
-
 		return $fields;
 	}
 
@@ -131,6 +126,7 @@ class Blog extends Page {
 	}
 
 
+
 	/**
 	 * This sets the title for our gridfield
 	 *
@@ -140,6 +136,17 @@ class Blog extends Page {
 		return _t('Blog.LumberjackTitle', 'Blog Posts');
 	}
 
+
+
+	/**
+	 * This overwrites lumberjacks default gridfield config.
+	 *
+	 * @return GridFieldConfig
+	 */
+	public function getLumberjackGridFieldConfig() {
+		return GridFieldConfig_Blog::create();
+	}
+
 }
 
 
diff --git a/composer.json b/composer.json
index f8694ca..01ea379 100755
--- a/composer.json
+++ b/composer.json
@@ -5,7 +5,7 @@
     "type": "silverstripe-module",
     "require": {
         "silverstripe/cms": "^3.1.0",
-	    "micmania1/silverstripe-lumberjack": "^1.0.0"
+	    "micmania1/silverstripe-lumberjack": "~1.1"
     },
     "license": "BSD-2-Clause",
     "authors": [