Fixed by adding missing code to file