Revert "Fix wrong type"

This reverts commit 6e536cb76e112b734d11f7c062f580abd335ddfa.