Added missing require and fixing small mistake