Fix visibility problem from last commit