Fix restore styles