Fix an important include