Update Unit Tests for Imaging