fixed `must return a value error`