fix running unit tests