diff --git a/dwave/gate/simulator/operation_generation.py b/dwave/gate/simulator/operation_generation.py
index dc4c55f..882e570 100644
--- a/dwave/gate/simulator/operation_generation.py
+++ b/dwave/gate/simulator/operation_generation.py
@@ -523,6 +523,7 @@ def generate_op(
     cython_file = (
         """\
 # distutils: language = c++
+# distutils: include_dirs = dwave/gate/simulator/
 # cython: language_level=3
 # THIS FILE WAS AUTOMATICALLY GENERATED BY dwave/gate/simulator/operation_generation.py
 
diff --git a/setup.cfg b/setup.cfg
index c412582..b3a74bd 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -10,6 +10,7 @@ classifiers =
     Programming Language :: Python :: 3.8
     Programming Language :: Python :: 3.9
     Programming Language :: Python :: 3.10
+    Programming Language :: Python :: 3.11
     Programming Language :: Python :: 3 :: Only
 
 long_description = file: README.rst
diff --git a/setup.py b/setup.py
index 6b883a9..3b2b3e7 100644
--- a/setup.py
+++ b/setup.py
@@ -46,30 +46,17 @@ def build_extensions(self):
         build_ext.build_extensions(self)
 
 
-extensions = [
-    Extension(
-        name="dwave.gate.simulator.simulator",
-        sources=["dwave/gate/simulator/simulator.pyx"],
-        include_dirs=["dwave/gate/simulator/"],
-        language="c++",
-    ),
-    Extension(
-        name="dwave.gate.registers.cyregister",
-        sources=["dwave/gate/registers/cyregister.pyx"],
-        include_dirs=["dwave/gate/registers/"],
-        language="c++",
-    ),
-]
-
 setup(
     name="dwave-gate",
     install_requires=[
         "numpy",
     ],
     ext_modules=cythonize(
-        extensions, annotate=bool(os.getenv("CYTHON_ANNOTATE", False))
+        ["dwave/gate/simulator/simulator.pyx",
+         "dwave/gate/registers/cyregister.pyx"
+         ],
+        annotate=bool(os.getenv("CYTHON_ANNOTATE", False))
     ),
     include_dirs=[numpy.get_include()],
     cmdclass={"build_ext": build_ext_compiler_check},
-    zip_safe=False,
 )