Fix for incorrect data type