Fixed dialog not closing

Dialog now closes on “OK” click()