Exactly that was my thinking as well. Pass the whole entity. :-)