Temp fix for server-only class being called on the server