Fix forgotten includes.