Remove registration package from code
Fixing import