Fixes bug found by CI