Fixed: added missing include header