Fix for ruby v2