minor import fix and debug print.