Tweak test setup to run on activerecord 4.1.0 as well