Director saw possible signs of Bruce Willis' aphasia diagnosis on set