Moved setKeepCallbackAsBool call to the correct plugin action