fix for bug