minor fix on previous commit