Trying to fix string comparison