bugfix the callback behaviors.

bugfix the callback behaviors. #81

and add tests for plain models callback.