minor fix in inital code