Bad import statement fixed.

fix #538