Pull request did not compile due to missing imports. Fixed the issue