Fix missing code exit

Title says all