Darkmode error di javascript.

Disini saya baru selesai membuat darkmode pada website, setelah lihat di channel youtube sekolah koding.

Jadi error di sini adalah ketika saya klik tombol dark berhasil dan ketika saya klik tombol light juga berhasil, tapi ketika saya refresh dalam mode light, tiba" backgroundnya otomatis kembali ke dark lagi, padahal saya belum klik tombol dark.

Kode Javascript : 

if (localStorage.getItem('theme') === 'dark')
	setDarkMode(true)

function setDarkMode(isDark) {
	if(isDark){
		document.body.setAttribute('class', 'darkmode')
		localStorage.setItem('theme', 'dark')
	} else {
		document.body.setAttribute('class','')
		localStorage.removeItem('class')
	}
}

Kode html :

<button onclick="setDarkMode(false)">Light</button> / <button onclick="setDarkMode(true)">Dark</button>

Maaf, udah berhasil.
Ternyata kesalahan nya ada di :

localStorage.removeItem('class')

Seharusnya :

localStorage.removeItem('theme')