Fixed som errors from last commit