https://github.com/amplab/MLI/issues/4 fix