Trying to correct transition bug