Fix gem bug