From 7464bd2521b3f5014dd8ddbd583509fb0153a1ff Mon Sep 17 00:00:00 2001
From: Mike Cao <mike@mikecao.com>
Date: Sun, 2 Apr 2023 17:28:25 -0700
Subject: [PATCH] Fixed types.

---
 lib/query.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/lib/query.ts b/lib/query.ts
index 0fa50595..4ce18b09 100644
--- a/lib/query.ts
+++ b/lib/query.ts
@@ -1,6 +1,6 @@
 import cache from 'lib/cache';
 import { getWebsite, getSession, getUser } from 'queries';
-import { Website } from './types';
+import { User, Website, Session } from '@prisma/client';
 
 export async function loadWebsite(websiteId: string): Promise<Website> {
   let website;
@@ -18,7 +18,7 @@ export async function loadWebsite(websiteId: string): Promise<Website> {
   return website;
 }
 
-export async function loadSession(sessionId: string): Promise<Website> {
+export async function loadSession(sessionId: string): Promise<Session> {
   let session;
 
   if (cache.enabled) {
@@ -34,7 +34,7 @@ export async function loadSession(sessionId: string): Promise<Website> {
   return session;
 }
 
-export async function loadUser(userId: string): Promise<Website> {
+export async function loadUser(userId: string): Promise<User> {
   let user;
 
   if (cache.enabled) {