Fix missing "return"