Small fix for tests
1.0.4