1 files changed, 7 insertions, 5 deletions
diff --git a/extensions/hosts.configure b/extensions/hosts.configure
index 6b068d04..11fcf573 100755
--- a/extensions/hosts.configure
+++ b/extensions/hosts.configure
@@ -22,27 +22,29 @@ import os
 import sys
 import socket
 
-import morphlib
+import writeexts
 
 def validate(var, line):
     xs = line.split()
     if len(xs) == 0:
-        raise morphlib.Error("`%s: %s': line is empty" % (var, line))
+        raise writeexts.ExtensionError(
+            "`%s: %s': line is empty" % (var, line))
 
     ip = xs[0]
     hostnames = xs[1:]
 
     if len(hostnames) == 0:
-        raise morphlib.Error("`%s: %s': missing hostname" % (var, line))
+        raise writeexts.ExtensionError(
+            "`%s: %s': missing hostname" % (var, line))
 
     family = socket.AF_INET6 if ':' in ip else socket.AF_INET
 
     try:
         socket.inet_pton(family, ip)
     except socket.error:
-        raise morphlib.Error("`%s: %s' invalid ip" % (var, ip))
+        raise writeexts.ExtensionError("`%s: %s' invalid ip" % (var, ip))
 
 envvars = {k: v for (k, v) in os.environ.iteritems() if k.startswith('HOSTS_')}
 
 conf_file = os.path.join(sys.argv[1], 'etc/hosts')
-morphlib.util.write_from_dict(conf_file, envvars, validate)
+writeexts.write_from_dict(conf_file, envvars, validate)