fix the return statements(2)