Fixes #28 'visibility buttons not working'

Fixed #28