fixed the minor fix