Fixes some broken test