Commad "rake what_column:add" breaks with message: "wrong constant name <"

[galetahub]

If ruby class looks like this:
class Test < ActiveRecord::Base

some code

class << self
def test_method
# some code
end
end
end

Commad "rake what_column:add" breaks with message: "wrong constant name <", becouse it wrong parse line "class << self".

So, please, replace regex "/class (.*)</" to "/class (\w.+)</".