Fixing the local imports.

Either the import path should be fully qualified or local.
Switching to local for now.