[MINOR] Minor integration issues