Fix up tests

3 files changed, 11 insertions, 10 deletions

diff --git a/distbuild/jm.py b/distbuild/jm.py
index 1277b4db..5592d5bd 100644
--- a/distbuild/jm.py
+++ b/distbuild/jm.py
@@ -20,10 +20,10 @@ import json
 import logging
 import yaml
 
-from sm import StateMachine 
-from stringbuffer import StringBuffer
-from sockbuf import (SocketBuffer, SocketBufferNewData, 
-                     SocketBufferEof)
+from distbuild.sm import StateMachine 
+from distbuild.stringbuffer import StringBuffer
+from distbuild.sockbuf import (SocketBuffer, SocketBufferNewData,
+                               SocketBufferEof)
 
 
 class JsonNewMessage(object):
diff --git a/distbuild/sockbuf.py b/distbuild/sockbuf.py
index fc0315b0..2feb7f2a 100644
--- a/distbuild/sockbuf.py
+++ b/distbuild/sockbuf.py
@@ -46,10 +46,10 @@ been emptied.
 '''
 
 
-from socketsrc import (SocketError, SocketReadable, SocketWriteable,
-                       SocketEventSource)
-from sm import StateMachine
-from stringbuffer import StringBuffer
+from distbuild.socketsrc import (SocketError, SocketReadable, SocketWriteable,
+                                 SocketEventSource)
+from distbuild.sm import StateMachine
+from distbuild.stringbuffer import StringBuffer
 
 
 class SocketBufferNewData(object):
diff --git a/distbuild/sockserv.py b/distbuild/sockserv.py
index 964352e7..d65f2008 100644
--- a/distbuild/sockserv.py
+++ b/distbuild/sockserv.py
@@ -18,8 +18,9 @@
 
 import logging
 
-from sm import StateMachine
-from socketsrc import NewConnection, SocketError, ListeningSocketEventSource
+from distbuild.sm import StateMachine
+from distbuild.socketsrc import (NewConnection, SocketError,
+                                 ListeningSocketEventSource)
 
 
 class _Close(object):