diff --git a/collects/typed-scheme/infer/infer.ss b/collects/typed-scheme/infer/infer.ss
index 01afd445..208943a3 100644
--- a/collects/typed-scheme/infer/infer.ss
+++ b/collects/typed-scheme/infer/infer.ss
@@ -3,25 +3,10 @@
 (require (except-in "../utils/utils.ss" infer))
 (require "infer-unit.ss" "constraints.ss" "dmap.ss" "signatures.ss"
          "restrict.ss" "promote-demote.ss"
-         scheme/contract
-         (rep type-rep)
+         (only-in scheme/unit provide-signature-elements)
          (utils unit-utils))
-
-(provide/contract
- [infer (((listof symbol?) (listof Type?) (listof Type?) (or/c (one-of/c #f) Type?) (listof symbol?)) 
-         ((or/c (one-of/c #f) Type?))
-         . ->* . 
-         (listof (list/c symbol? Type?)))]
- [infer/vararg (((listof symbol?) (listof Type?) (listof Type?) Type? (or/c (one-of/c #f) Type?) (listof symbol?)) 
-                ((or/c (one-of/c #f) Type?))
-                . ->* . 
-                (listof (list/c symbol? Type?)))]
- [infer/dots (((listof symbol?) symbol? (listof Type?) (listof Type?) Type? (or/c (one-of/c #f) Type?) (listof symbol?)) 
-              (#:expected (or/c (one-of/c #f) Type?))
-              . ->* . 
-              (listof (list/c symbol? Type?)))])
          
-(provide restrict)
+(provide-signature-elements restrict^ infer^)
 
 (define-values/link-units/infer
-  infer@ constraints@ dmap@ restrict@ promote-demote@)
+  infer@ constraints@ dmap@ restrict@ promote-demote@)
\ No newline at end of file