Fixed class not being public