Fix data type warning