diff --git a/common/pybindings.cc b/common/pybindings.cc
index cc4ad006..3a43478c 100644
--- a/common/pybindings.cc
+++ b/common/pybindings.cc
@@ -108,7 +108,7 @@ BOOST_PYTHON_MODULE(MODULE_NAME)
     WRAP_MAP(decltype(CellInfo::ports), "IdPortMap");
     // WRAP_MAP(decltype(CellInfo::pins), "IdIdMap");
 
-    class_<Context, Context *>("Context", no_init)
+    class_<BaseCtx, BaseCtx *>("BaseCtx", no_init)
             .def_readwrite("nets", &Context::nets)
             .def_readwrite("cells", &Context::cells);
 
@@ -123,6 +123,8 @@ BOOST_PYTHON_MODULE(MODULE_NAME)
             .def(self < self)
             .def(self == self);
     arch_wrap_python();
+
+    class_<Context, Context *, bases<Arch>>("Context", no_init).def("checksum", &Context::checksum);
 }
 
 void arch_appendinittab() { PyImport_AppendInittab(TOSTRING(MODULE_NAME), PYINIT_MODULE_NAME); }
@@ -179,4 +181,4 @@ void execute_python_file(const char *python_file)
 
 NEXTPNR_NAMESPACE_END
 
-#endif // NO_PYTHON
\ No newline at end of file
+#endif // NO_PYTHON
diff --git a/ice40/pybindings.cc b/ice40/pybindings.cc
index 4c707e65..bb591289 100644
--- a/ice40/pybindings.cc
+++ b/ice40/pybindings.cc
@@ -53,7 +53,7 @@ void arch_wrap_python()
             ;
 #undef X
 
-    class_<Arch>("Arch", init<ArchArgs>())
+    class_<Arch, Arch *, bases<BaseCtx>>("Arch", init<ArchArgs>())
             .def("getBelByName", &Arch::getBelByName)
             .def("getWireByName", &Arch::getWireByName)
             .def("getBelName", &Arch::getBelName)
@@ -85,4 +85,4 @@ void arch_wrap_python()
 
 NEXTPNR_NAMESPACE_END
 
-#endif // NO_PYTHON
\ No newline at end of file
+#endif // NO_PYTHON