fixiing 'code' bug