fixed minor locle issue