Changed private to protected in Units class