Well, the movie does sort of gloss over these facts. However, since Vito was shot and Michael was his son, it wouldn't have taken a genius to connect the dots when McCluskey was killed. I believe that the waiter had identified Michael's picture as well.