fix NB errors
fix NB errors