Fix NotSerializableException