Updated dialogTest to work with framework