fix for code