Add a missing `return` statement