Fixed screen duration issue