Object diagrama fix