additional code to fix the issue