Fix integration tests.

Arsen, I know you are going to kill me for this, but I repeated the same mistake again. I have added closed #15 in my last commit message.