fixing backets error

Fixing brackets after last commit :)