Fix issue #6

Signed-off-by: Janet Jin <janet.jin@turn.com>