fix tests
Merge pull request #10 from Parcela/parameter-shift

fix tests