fixed 'no display attached'
quick fix cleanup, fixes #2