Fixing missing code