Fixed issue on custom dialog builder