Fixup dockerfile to account for recent changes.
Merge pull request #78 from mrunalp/fix_dockerfile

Fixup dockerfile to account for recent changes.