fix funciton return