Quick hack to make it work with ruby 2.0