Fix test include