Style correction
Fixed missing include