minor import fix and debug print.
minor import fix and debug print.