Added missing include (thanks to Sebastian Ordas)