diff --git a/php/modules/album.php b/php/modules/album.php
index 66c6639..05a87f0 100755
--- a/php/modules/album.php
+++ b/php/modules/album.php
@@ -298,7 +298,7 @@ function setAlbumPublic($albumID, $password) {
 
 	$result	= $database->query("SELECT public FROM lychee_albums WHERE id = '$albumID';");
 	$row	= $result->fetch_object();
-	$public	= ($row->public===0 ? 1 : 0);
+	$public	= ($row->public=='0' ? 1 : 0);
 	
 	$result = $database->query("UPDATE lychee_albums SET public = '$public', password = NULL WHERE id = '$albumID';");
 	if (!$result) return false;