Fix test...duh