Yes, so I have changed it now. Made the function check for both versions' strings and return accordingly