Fix by hooking unmount.