Fix callback parameter to pass into setUp