Fixed mismatched type issue