Fix for dialog