cleanup unit test
Merge pull request #2 from agilemedialab/unit_test_fix

cleanup unit test