Fix viewport UI