add proper display to fix tests