Fixed Bug in ViewController