Updating ksp tests since we dropped the wrapper function.