Fixed last commit cause the PR was lazy