Fix get_display import error

This should fix the cyclic import issues being caused by ```get_display```.