Fix gem ut bug