Work with Rails > 3 and < 4