Fixed missing headers error