Made minor fix