Fix: Change output data type