A simpler explanation yeah. Plus Johngalli actually worked with Pucci in-depth and knew about his Stand so if his memory were erased he might notice gaps and/or break allegiance with him realizing he'd been messed with. Either way it'd be cleaner for him to die especially since Pucci figured he'd no longer have use for him with Jotaro dealt with.