Small fix

Compiles for me!
:thumbsup: