Fixed return statement, only a single statement returned