Bug should expand bug

This should fix a bug where the shouldExpand flag doesn't behave as it should due to the child views not being measured after updating their layout params (unless I've been doing something wrong...).