fixed call to nn_device

"wrapper" was missing