Fix input sizes