Fix CI

Fixes #1 #2