fixed code issue