Fixing display code from example.