fix input sizes
fix input sizes