add missing dependency
make it somehow python2 compatible