Selenium on Rails breaks on Rails 2.1

We upgraded ourselves to Rails 2.1. And builds started failing. Bamboo showed me a “undefined method – register_template_handler” error.

.../​vendor/​ plugins/​selenium-on-rails/​lib/​selenium_on_rails/​selenese.rb:3: undefined method `register_template_handler' for ActionView::Base:Class (NoMethodError)

It happened for rselenese.rb too.
It seems they moved “register_template_handler” from ActionView::Base to ActionView::Template in Rails 2.1. So I went in and changed

ActionView::Base.register_template_handler 'sel', SeleniumOnRails::Selenese
To
ActionView::Template.register_template_handler 'sel', SeleniumOnRails::Selenese


And bingo! it worked fine as it was working earlier.