Fix bug in input size