Updated imports with the new custom layer package name

Updated imports with the new custom layer package name #820