Fixed code file reference issue