fixed my bug