Fixed unit test
Updating to v 1.0.4