Corrected import error (fix #4)