fix return object
fix display bug