Fix with node module changes

This will fix #2