Fixed malloc error