Fix code after pull request