Fix for duration display issue