fixed broken function
fix