diff --git a/tests/GridFieldAddNewMultiClassTest.php b/tests/GridFieldAddNewMultiClassTest.php
index 33e6943..10e476e 100644
--- a/tests/GridFieldAddNewMultiClassTest.php
+++ b/tests/GridFieldAddNewMultiClassTest.php
@@ -3,7 +3,7 @@
 use SilverStripe\Dev\SapphireTest;
 use SilverStripe\Dev\TestOnly;
 use SilverStripe\Forms\GridField\GridField;
-use SilverStripe\GridFieldExtensions\GridFieldAddNewMultiClass;
+use SilverStripeAustralia\GridFieldExtensions\GridFieldAddNewMultiClass;
 
 /**
  * Tests for {@link GridFieldAddNewMultiClass}.
diff --git a/tests/GridFieldOrderableRowsTest.php b/tests/GridFieldOrderableRowsTest.php
index 6343a72..313e1a7 100644
--- a/tests/GridFieldOrderableRowsTest.php
+++ b/tests/GridFieldOrderableRowsTest.php
@@ -4,15 +4,14 @@ use SilverStripe\Dev\SapphireTest;
 use SilverStripe\Dev\TestOnly;
 use SilverStripe\Forms\GridField\GridField;
 use SilverStripe\Forms\GridField\GridFieldConfig_RelationEditor;
-use SilverStripe\GridFieldExtensions\GridFieldOrderableRows;
 use SilverStripe\ORM\DataObject;
+use SilverStripeAustralia\GridFieldExtensions\GridFieldOrderableRows;
 
 /**
  * Tests for the {@link GridFieldOrderableRows} component.
  */
 class GridFieldOrderableRowsTest extends SapphireTest
 {
-
     protected $usesDatabase = true;
 
     // protected static $fixture_file = 'GridFieldOrderableRowsTest.yml';