Added support for testing with ActiveRecord