Fixed broken test
Fixed broken test