Fixed broken test...