Fixed image regression introduced in the refactor.