Quick bug fix. Stuck on performance benchmark