Updating to support rails version again