forgot to update main class property