Fix Rails 3 support by calling the appropriate module