Fixed broken test