fix whole training screen button bug