Dialog Box fix..