From 774355d608aab5265f91a337956dedc3b2565c3b Mon Sep 17 00:00:00 2001
From: Dolan <dolan_miu@hotmail.com>
Date: Wed, 31 Jan 2018 01:23:13 +0000
Subject: [PATCH] Fix tests

---
 src/file/document/document.spec.ts | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/file/document/document.spec.ts b/src/file/document/document.spec.ts
index a3981febaa..99285fef21 100644
--- a/src/file/document/document.spec.ts
+++ b/src/file/document/document.spec.ts
@@ -1,13 +1,15 @@
 import { assert, expect } from "chai";
 
-import * as docx from "../../";
 import { Formatter } from "../../export/formatter";
+import { Paragraph } from "../paragraph";
+import { Table } from "../table";
+import { Document } from "./document";
 
 describe("Document", () => {
-    let document: docx.Document;
+    let document: Document;
 
     beforeEach(() => {
-        document = new docx.Document();
+        document = new Document();
     });
 
     describe("#constructor()", () => {
@@ -27,7 +29,7 @@ describe("Document", () => {
     describe("#createParagraph", () => {
         it("should create a new paragraph and append it to body", () => {
             const para = document.createParagraph();
-            expect(para).to.be.an.instanceof(docx.Paragraph);
+            expect(para).to.be.an.instanceof(Paragraph);
             const body = new Formatter().format(document)["w:document"][1]["w:body"];
             expect(body)
                 .to.be.an("array")
@@ -37,7 +39,7 @@ describe("Document", () => {
 
         it("should use the text given to create a run in the paragraph", () => {
             const para = document.createParagraph("sample paragraph text");
-            expect(para).to.be.an.instanceof(docx.Paragraph);
+            expect(para).to.be.an.instanceof(Paragraph);
             const body = new Formatter().format(document)["w:document"][1]["w:body"];
             expect(body)
                 .to.be.an("array")
@@ -53,7 +55,7 @@ describe("Document", () => {
     describe("#createTable", () => {
         it("should create a new table and append it to body", () => {
             const table = document.createTable(2, 3);
-            expect(table).to.be.an.instanceof(docx.Table);
+            expect(table).to.be.an.instanceof(Table);
             const body = new Formatter().format(document)["w:document"][1]["w:body"];
             expect(body)
                 .to.be.an("array")