Added missing include (was failing to compile in OSX).