Changes in test case due to colorbar no longer being axis