fixed a bug that impacted nn?