Fix colouring of widget

Fix as suggested by jbarket in issue #1