OOP fixes