Fix ruby code. Python version needs update