diff --git a/src/firebase/firestore.ts b/src/firebase/firestore.ts
index 21b93fb..e40c495 100644
--- a/src/firebase/firestore.ts
+++ b/src/firebase/firestore.ts
@@ -218,8 +218,8 @@ export function saveEntity(
  * Delete an entity
  * @param entity
  */
-export function deleteEntity(
-    entity: Entity<any>
+export function deleteEntity<S extends EntitySchema>(
+    entity: Entity<S>
 ): Promise<void> {
     console.debug("Deleting entity", entity);
     return entity.reference.delete();