Fix switch theme

Without this it was not working for me.