Minor bug fix, missing imported library.