Change code from PHPStorm