Fixed broken test with 1.0.2