Fixing trainer dialog bug when trainer takes too long to load.