Small fix for console screen input