Fix finish button error