Add check for view controller