fixed bug alittlle bit