Minor change to listen function