Removes errant print statement and fixes import error

Signed-off-by: Oren Leiman <oren@neutral.io>