Fix import error.
Release 1.6.1.