Fix after theme chooser

Change-Id: Ia445acba0a48ebe3e6672505fb038545e74c97d2
Signed-off-by: iceandfire <arhamjamal@gmail.com>