add type checking in tests