Fix test was broken