Fix build
Add pub shied