Fix wrong argument type