Fix error in nmj.py module