diff --git a/UI/media-controls.cpp b/UI/media-controls.cpp
index 0248ade81..c73261d21 100644
--- a/UI/media-controls.cpp
+++ b/UI/media-controls.cpp
@@ -253,6 +253,7 @@ void MediaControls::RefreshControls()
 	switch (state) {
 	case OBS_MEDIA_STATE_STOPPED:
 	case OBS_MEDIA_STATE_ENDED:
+	case OBS_MEDIA_STATE_NONE:
 		SetRestartState();
 		break;
 	case OBS_MEDIA_STATE_PLAYING: