Fix wrong type hint