Work with entity NS

Signed-off-by: Alexander Shepetko <a@shepetko.com>