Fix errors

CodeSniffer errors found - Fix