Fix bug portion of PLUGIN-58.

Signed-off-by: Ian Macalinao <ianmacalinao@gmail.com>