Fix size-classes

Fix size-classes