Fix type issue