Change import to absolute import

Fixes imports for python 3 (and prevents any funnies in 2 too).