fixed smalll bug