diff --git a/pysages/backends/lammps.py b/pysages/backends/lammps.py
index 37508ff8..b54ec62c 100644
--- a/pysages/backends/lammps.py
+++ b/pysages/backends/lammps.py
@@ -83,6 +83,11 @@ def update(timestep):
 
         self.set_callback(update)
 
+        def virial_from_method(virial):
+            #virial = sampling_method.get_virial()
+
+        self.set_virial_callback(virial_from_method)
+
     def _partial_snapshot(self, include_masses: bool = False):
         positions = from_dlpack(dlext.positions(self.view, self.location))
         types = from_dlpack(dlext.types(self.view, self.location))