Fixed broken test...blah