Fix problem with code reference