diff --git a/server/extension/sql/200_quotas.sql b/server/extension/sql/200_quotas.sql
index 0b57781..58f49c6 100644
--- a/server/extension/sql/200_quotas.sql
+++ b/server/extension/sql/200_quotas.sql
@@ -59,13 +59,15 @@ $$ LANGUAGE plpythonu;
 CREATE OR REPLACE FUNCTION cdb_dataservices_server.cdb_enough_quota(
   username TEXT,
   orgname TEXT,
-  service TEXT,
+  service_ cdb_dataservices_server.service_type,
   input_size NUMERIC)
 returns BOOLEAN AS $$
   DECLARE
     params cdb_dataservices_server.service_params;
   BEGIN
-    SELECT * INTO params FROM cdb_dataservices_server.cdb_service_params(username, orgname, service);
+    SELECT * INTO params
+      FROM cdb_dataservices_server.cdb_service_params(username, orgname) AS p
+      WHERE p.service = service_;
     RETURN params.soft_limit OR ((params.used_quota + input_size) <= params.monthly_quota);
   END
 $$ LANGUAGE plpgsql;