Missing includes fix for linux