2 files changed, 11 insertions, 11 deletions
diff --git a/src/site/components/album/AlbumDetails.vue b/src/site/components/album/AlbumDetails.vue
index 10925df..81819b2 100644
--- a/src/site/components/album/AlbumDetails.vue
+++ b/src/site/components/album/AlbumDetails.vue
@@ -100,9 +100,7 @@
 					<div class="level-right">
 						<div class="level-item">
 							<b-switch
-								v-model="isNsfw"
-								:false-value="0"
-								:true-value="1"
+								:value="nsfw"
 								@input="toggleNsfw()" />
 						</div>
 						<div class="level-item">
@@ -132,6 +130,10 @@ export default {
 		details: {
 			'type': Object,
 			'default': () => ({})
+		},
+		nsfw: {
+			'type': Boolean,
+			'default': false
 		}
 	},
 	data() {
@@ -141,10 +143,7 @@ export default {
 		};
 	},
 	computed: {
-		...mapState(['config', 'auth']),
-		isNsfw() {
-			return this.$store.state.albums.list.find(a => a.id === this.albumId).nsfw;
-		}
+		...mapState(['config', 'auth'])
 	},
 	mounted() {
 		console.log(this.isNsfw);
@@ -156,7 +155,7 @@ export default {
 			updateLinkOptionsAction: 'albums/updateLinkOptions',
 			createLinkAction: 'albums/createLink',
 			createCustomLinkAction: 'albums/createCustomLink',
-			toggleNsfw: 'albums/toggleNsfw',
+			toggleNsfwAction: 'albums/toggleNsfw',
 			alert: 'alert/set'
 		}),
 		promptDeleteAlbum(id) {
@@ -217,9 +216,9 @@ export default {
 		},
 		async toggleNsfw() {
 			try {
-				const response = await this.toggleNsfw({
+				const response = await this.toggleNsfwAction({
 					albumId: this.albumId,
-					nsfw: !this.isNsfw
+					nsfw: !this.nsfw
 				});
 				this.alert({ text: response.message, error: false });
 			} catch (e) {
diff --git a/src/site/components/album/AlbumEntry.vue b/src/site/components/album/AlbumEntry.vue
index b8a5cb2..8947fa5 100644
--- a/src/site/components/album/AlbumEntry.vue
+++ b/src/site/components/album/AlbumEntry.vue
@@ -53,7 +53,8 @@
 		<AlbumDetails
 			v-if="isExpanded"
 			:details="getDetails(album.id)"
-			:album-id="album.id" />
+			:album-id="album.id"
+			:nsfw="album.nsfw" />
 	</div>
 </template>