change setup and teardown callback type (#1934)

change setup and teardown callback type (#1934) #2611