Fixed compile error

#4 introduced this