Added tests for callback