Fixed small error for main view