[~TASK] Fixed bugs