Fix missing include and remove debug prints