fix missing include and return value