Change parameter type

There is a compilation warning generated by a discrepancy in the parameter type between the .h and .m.