Fix visibility of class