Fix type warning