Fix issue with missing gem