Fixed missing test, and images