Improved tests coverage.

https://gist.github.com/przemekmazurek/599aa13c170c04fd330d