Fix direct message function