Try to fix running unit tests