diff --git a/atlassian-connect-descriptor.cabal b/atlassian-connect-descriptor.cabal
--- a/atlassian-connect-descriptor.cabal
+++ b/atlassian-connect-descriptor.cabal
@@ -10,7 +10,7 @@
 -- PVP summary:      +-+------- breaking API changes
 --                   | | +----- non-breaking API additions
 --                   | | | +--- code changes with no API change
-version:             0.4.4.0
+version:             0.4.4.1
 
 -- A short (one-line) description of the package.
 synopsis:            Code that helps you create a valid Atlassian Connect Descriptor.
@@ -92,7 +92,7 @@
   ghc-options: -Wall
 
 Test-Suite test-descriptor
-  hs-source-dirs: tests, src
+  hs-source-dirs: tests
   type:        detailed-0.9
 
   test-module: Test
@@ -102,18 +102,10 @@
                  , ConditionsTest
                  , ModulesTest
                  , DescriptorTest
-                 , Data.Connect.Descriptor
-                 , Data.Connect.AesonHelpers
-                 , Data.Connect.BaseTypes
-                 , Data.Connect.Conditions
-                 , Data.Connect.Lifecycle
-                 , Data.Connect.Modules
-                 , Data.Connect.OrphanInstances
-                 , Data.Connect.Scopes
-                 , Data.Connect.Webhooks
 
   build-depends:
     base
+    , atlassian-connect-descriptor
     , Cabal >= 1.20
     , HUnit
     , vector
diff --git a/tests/ConditionsTest.hs b/tests/ConditionsTest.hs
--- a/tests/ConditionsTest.hs
+++ b/tests/ConditionsTest.hs
@@ -4,7 +4,7 @@
 import           AssertionHelpers
 import           Control.Monad
 import           Data.Aeson
-import           Data.Connect.Conditions
+import           Data.Connect.Descriptor
 import qualified Data.Text               as T
 import           Test.HUnit
 import           ValueExtractors
diff --git a/tests/ModulesTest.hs b/tests/ModulesTest.hs
--- a/tests/ModulesTest.hs
+++ b/tests/ModulesTest.hs
@@ -4,10 +4,7 @@
 import           AssertionHelpers
 import           Control.Applicative
 import           Data.Aeson
-import           Data.Connect.BaseTypes
-import           Data.Connect.Conditions (JIRACondition (..),
-                                          staticJiraCondition)
-import           Data.Connect.Modules
+import           Data.Connect.Descriptor
 import qualified Data.HashMap.Strict     as HM
 import           Test.HUnit
 import           ValueExtractors