[fix] Postpone type checking of invariants, axioms, init-conditions and pure functions

[fix] Postpone type checking of invariants, axioms, init-conditions and pure functions #1440