after_load and after_find now accept entity instead of criteria