Fix refactor leftover