Fixed Pr/2

..and apparently committed into branch instead of PR. wtf? :)