Fix CI

With the new refactor !83 (merged), the CI is completely broken. This needs to be fixed