Fix failing tests

tests were failing as the implementation of StylePicker changed so the method for testing needed updating.