A system and method. The system may include a display, a lens having distortion, an image generator, and a processor. The lens may be configured to focus light received from an environment. The image generator may be configured to receive the light from the lens and output a stream of images as image data, wherein each of the stream of images is distorted. The processor may be configured to: receive the image data from the image generator; detect a point source object in the stream of images of the image data; enhance the point source object in the stream of images of the image data; undistort the stream of images of the image data having an enhanced point source object; and output a stream of undistorted images as undistorted image data to the display.