diff --git a/README.md b/README.md
index cb52a86..adec701 100644
--- a/README.md
+++ b/README.md
@@ -15,7 +15,8 @@ For convenience, a high level API has been added:
 ```
 import getRdfaGraph from graph-rdfa-processor;
 
-let graph = getRdfaGraph(document, opts);
+let opts = {baseURI: 'http://example.com'};
+let graph = getRdfaGraph(documentOrNode, opts);
 console.log(graph.toString());
 ```
 
diff --git a/package.json b/package.json
index ba8d13d..7b39bdb 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "graph-rdfa-processor",
-  "version": "1.0.1",
+  "version": "1.1.0",
   "description": "Green turtle GraphRdfaProcessor extracted and available as a commonJS module",
   "main": "dist/index.js",
   "scripts": {
diff --git a/src/index.js b/src/index.js
index 96f6488..46af010 100644
--- a/src/index.js
+++ b/src/index.js
@@ -3,7 +3,8 @@ import GraphRDFaProcessor from './graph-rdfa-processor';
 import { RDFaGraph } from './rdfa-graph';
 
 export default function(document, options = {}) {
-  let baseURI = options.baseURI ? options.baseURI : document.documentElement.baseURI;
+  let node = document.documentElement || document;
+  let baseURI = options.baseURI ? options.baseURI : node.baseURI;
 
   let graph = new RDFaGraph();
 
@@ -13,6 +14,6 @@ export default function(document, options = {}) {
   };
 
   var processor = new GraphRDFaProcessor(target);
-  processor.process(document.documentElement, options);
+  processor.process(node, options);
   return target.graph;
 };
diff --git a/src/rdfa-processor.js b/src/rdfa-processor.js
index 2b918e1..eab295a 100644
--- a/src/rdfa-processor.js
+++ b/src/rdfa-processor.js
@@ -278,7 +278,6 @@ export default class RDFaProcessor extends URIResolver {
 
 
   process(node, options) {
-
     if (node.nodeType==Node.DOCUMENT_NODE) {
       node = node.documentElement;
       this.setContext(node);
diff --git a/test/test.js b/test/test.js
index 23eb3d2..407f025 100644
--- a/test/test.js
+++ b/test/test.js
@@ -3,26 +3,34 @@ import assert from 'assert';
 import getRDFaGraph from '../src';
 
 describe('getRDFaGraph', function() {
-  it('should getRDFaGraph from a document', function() {
-    let html =
-    `<div typeof="rdfs:Class" resource="http://schema.org/CreativeWork">
+  let html =
+  `<div typeof="rdfs:Class" resource="http://schema.org/CreativeWork">
         <span class="h" property="rdfs:label">CreativeWork</span>
         <span property="rdfs:comment">The most generic kind of creative work, including books, movies, photographs, software programs, etc.</span>
         <span>Subclass of: <a property="rdfs:subClassOf" href="http://schema.org/Thing">Thing</a></span>
         <span>Source:  <a property="dc:source" href="http://www.w3.org/wiki/WebSchemas/SchemaDotOrgSources#source_rNews">rNews</a></span>
      </div>`;
 
-    let { document } = jsdom(html).defaultView.window;
-
-    let graph = getRDFaGraph(document, {baseURI: 'http://localhost'});
 
-    let expected = `<http://schema.org/CreativeWork> <http://www.w3.org/1999/02/22-rdf-syntax-ns#type> <http://www.w3.org/2000/01/rdf-schema#Class>;
+  let expected = `<http://schema.org/CreativeWork> <http://www.w3.org/1999/02/22-rdf-syntax-ns#type> <http://www.w3.org/2000/01/rdf-schema#Class>;
  <http://www.w3.org/2000/01/rdf-schema#label> "CreativeWork";
  <http://www.w3.org/2000/01/rdf-schema#comment> "The most generic kind of creative work, including books, movies, photographs, software programs, etc.";
  <http://www.w3.org/2000/01/rdf-schema#subClassOf> <http://schema.org/Thing>;
  <http://purl.org/dc/terms/source> <http://www.w3.org/wiki/WebSchemas/SchemaDotOrgSources#source_rNews> .
 `;
 
-   assert.equal(graph.toString(), expected);
+
+  it('should getRDFaGraph from a document', function() {
+    let { document } = jsdom(html).defaultView.window;
+    let graph = getRDFaGraph(document, {baseURI: 'http://localhost'});
+    assert.equal(graph.toString(), expected);
+  });
+
+
+  it('should getRDFaGraph from a node', function() {
+    let { document } = jsdom(html).defaultView.window;
+    let graph = getRDFaGraph(document.getElementsByTagName('div')[0], {baseURI: 'http://localhost'});
+    assert.equal(graph.toString(), expected);
   });
+
 });