Fixing tests
Releasing 0.5.1