fixed broken tests
Merge pull request #23 from mhshams/bug/rest

fixed broken tests