fixing issue with cross class communication