Made test passing
Merge pull request #2 from STRd6/update_test

Made test passing