diff --git a/waifu2x-caffe-gui/Source.cpp b/waifu2x-caffe-gui/Source.cpp
index 5c9acba..109bf74 100644
--- a/waifu2x-caffe-gui/Source.cpp
+++ b/waifu2x-caffe-gui/Source.cpp
@@ -434,6 +434,7 @@ public:
 
 		EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), TRUE);
 		EnableWindow(GetDlgItem(dh, IDC_BUTTON_EXEC), FALSE);
+		EnableWindow(GetDlgItem(dh, IDC_BUTTON_CHECK_CUDNN), FALSE);
 
 		SetWindowTextA(GetDlgItem(hWnd, IDC_EDIT_LOG), "");
 		logMessage.clear();
@@ -444,6 +445,7 @@ public:
 		processThread.join();
 		EnableWindow(GetDlgItem(dh, IDC_BUTTON_CANCEL), FALSE);
 		EnableWindow(GetDlgItem(dh, IDC_BUTTON_EXEC), TRUE);
+		EnableWindow(GetDlgItem(dh, IDC_BUTTON_CHECK_CUDNN), TRUE);
 
 		if (!isLastError)
 		{