Add better test that also works on CI