diff --git a/packages/client/src/hooks/use-websocket.ts b/packages/client/src/hooks/use-websocket.ts
index 1e90316..60cef78 100644
--- a/packages/client/src/hooks/use-websocket.ts
+++ b/packages/client/src/hooks/use-websocket.ts
@@ -28,6 +28,11 @@ export function useWebSocket(roomCode: string) {
 		addPlayer,
 		setAct,
 		reset,
+		setGameState,
+		addDish,
+		recordDishGuess,
+		lockPredictions,
+		setDishResults,
 	} = useRoomStore()
 
 	const send = useCallback((message: ClientMessage) => {
@@ -84,6 +89,21 @@ export function useWebSocket(roomCode: string) {
 				case "room_ended":
 					setAct("ended")
 					break
+				case "game_state":
+					setGameState(msg.gameState)
+					break
+				case "predictions_locked":
+					lockPredictions()
+					break
+				case "dish_added":
+					addDish(msg.dish)
+					break
+				case "dish_guess_recorded":
+					recordDishGuess(msg.dishId, msg.guessedCountry)
+					break
+				case "dishes_revealed":
+					setDishResults(msg.results)
+					break
 				case "error":
 					console.error("Server error:", msg.message)
 					break
@@ -98,7 +118,7 @@ export function useWebSocket(roomCode: string) {
 			ws.close()
 			reset()
 		}
-	}, [roomCode, setRoom, setMySessionId, setConnectionStatus, updatePlayerConnected, addPlayer, setAct, reset])
+	}, [roomCode, setRoom, setMySessionId, setConnectionStatus, updatePlayerConnected, addPlayer, setAct, reset, setGameState, addDish, recordDishGuess, lockPredictions, setDishResults])
 
 	return { send }
 }