Little fix input size etc.