Fix bug from refactor to make build pass again