diff --git a/src/Test/WebDriver/Commands/Internal.hs b/src/Test/WebDriver/Commands/Internal.hs
--- a/src/Test/WebDriver/Commands/Internal.hs
+++ b/src/Test/WebDriver/Commands/Internal.hs
@@ -3,38 +3,37 @@
 module Test.WebDriver.Commands.Internal
    ( clientScripts ) where
 
-import Control.Applicative
 import Data.Maybe (catMaybes)
 import Data.Monoid ((<>))
-import Language.JavaScript.Parser (JSNode(..), Node(..))
+import Language.JavaScript.Parser (JSAST(..), JSExpression(..), JSStatement(..))
 import qualified Data.HashMap.Lazy as M
 import qualified Data.Text as T
 import qualified Language.JavaScript.Parser as JS
 
 
 -- | Parse top level javascript commands
-parseClientTop :: JSNode -> [(T.Text,T.Text)]
-parseClientTop (NN (JSSourceElementsTop es)) = catMaybes $ map parseClientDef es
+parseClientTop :: JSAST -> [(T.Text,T.Text)]
+parseClientTop (JSAstProgram es _) = catMaybes $ map parseClientDef es
 parseClientTop _ = []
 
 -- | Build a function body that can be passed to async.  The body must use
 -- arguments array but the code from clientSideScripts.js uses function arguments.
-buildFunction :: Node -> T.Text
+buildFunction :: JSExpression -> T.Text
 buildFunction func = "return (" <> T.pack func' <> ").apply(null, arguments);"
     where
-        func' = dropWhile (=='\n') $ JS.renderToString $ NN func
+        func' = dropWhile (=='\n') $ JS.renderToString $ JSAstExpression func JS.JSNoAnnot
 
 -- | Parse a call clientSideScripts.somefunction = function() {...}, returning the function name
 -- and the body.
-parseClientDef :: JSNode -> Maybe (T.Text, T.Text)
-parseClientDef (NN (JSExpression
-                        [NN (JSMemberDot
-                            [NT (JSIdentifier "clientSideScripts") _ _]
-                            _ -- literal .
-                            (NT (JSIdentifier name) _ _))
-                        , NN (JSOperator (NT (JSLiteral "=") _ _))
-                        , NN (func@(JSFunctionExpression _ _ _ _ _ _))
-                        ]))
+parseClientDef :: JSStatement -> Maybe (T.Text, T.Text)
+parseClientDef (JSAssignStatement
+                    (JSMemberDot
+                        (JSIdentifier _ "clientSideScripts")
+                        _ -- annotation
+                        (JSIdentifier _ name))
+                    _ -- assign operator
+                    (func@(JSFunctionExpression{}))
+                    _) -- semicolon
     = Just (T.pack name, buildFunction func)
                         -- Use renderToString instead of renderJS so we don't need to depend on builder
 parseClientDef _ = Nothing
diff --git a/webdriver-angular.cabal b/webdriver-angular.cabal
--- a/webdriver-angular.cabal
+++ b/webdriver-angular.cabal
@@ -1,5 +1,5 @@
 name:              webdriver-angular
-version:           0.1.9
+version:           0.1.10
 cabal-version:     >= 1.8
 build-type:        Simple
 synopsis:          Webdriver actions to assist with testing a webpage which uses Angular.Js
@@ -42,7 +42,7 @@
 
                  , webdriver            >= 0.6  && < 0.9
                  , aeson                >= 0.6
-                 , language-javascript  >= 0.5  && < 0.6
+                 , language-javascript  >= 0.6  && < 0.7
                  , template-haskell     >= 0.6
                  , text                 >= 0.11
                  , transformers         >= 0.3