Cannot get comments within a build stage

[rcjsuen]

It seems like we only add comments to the Dockerfile but not an ImageTemplate.

import { DockerfileParser } from 'dockerfile-ast';
let dockerfile = DockerfileParser.parse("FROM scratch\n# comment\nRUN echo \"$VAR\"");
let image = dockerfile.getContainingImage({ line: 2, character: 1 });
console.log(image.getComments().length);

The above code returns 0 instead of 1. :(