app/models/album_photo.rb in ruby_gallery-0.1.5 vs app/models/album_photo.rb in ruby_gallery-0.1.6

@@ -1,6 +1,5 @@
 class AlbumPhoto < ActiveRecord::Base
   attr_accessible :photo
-  
-  belongs_to :album_attachment
+  belongs_to :album_photoable, polymorphic: true
   has_attached_file :photo, :styles => { :medium => "300x300>", :thumb => "100x100>" }
 end
\ No newline at end of file